hub / github.com/YunwenTechnology/QueryGeneration / get_shape_list

Function get_shape_list

train/utils.py:48–82 · view source on GitHub ↗

Returns a list of the shape of tensor, preferring static dimensions. Args: tensor: A tf.Tensor object to find the shape of. expected_rank: (optional) int. The expected rank of `tensor`. If this is specified and the `tensor` has a different rank, and exception will be

(tensor, expected_rank=None, name=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 13

_attention_projection_and_transposeFunction · 0.90

attention_layerFunction · 0.90

residual_mlp_layerFunction · 0.90

embedFunction · 0.90

_top_p_sampleFunction · 0.90

_top_k_sampleFunction · 0.90

__init__Method · 0.90

model_fnFunction · 0.90

sample_stepFunction · 0.90

initialize_from_contextFunction · 0.90

sampleFunction · 0.90

condFunction · 0.90

Calls 1

assert_rankFunction · 0.85

Tested by

no test coverage detected