MCPcopy Index your code
hub / github.com/YunwenTechnology/QueryGeneration / __init__

Method __init__

train/modeling.py:402–505  ·  view source on GitHub ↗

:param config: :param is_training: :param input_ids: Tensor thats of size [batch_size, seq_length] :param cache: Optionally, a tensor to use that will contain cached information of the size [batch_size, num_layers, 2, num_heads, cache_length, features]

(self,
                 config: GroverConfig,
                 is_training,
                 input_ids,
                 cache=None,
                 do_cache=False,
                 pad_token_id=0,
                 chop_off_last_token=True,
                 scope=None,
                 reuse=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 5

get_shape_listFunction · 0.90
get_attention_maskFunction · 0.90
embedFunction · 0.85
attention_layerFunction · 0.85
residual_mlp_layerFunction · 0.85

Tested by

no test coverage detected