Searched refs:value_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | ragged_to_dense_util.cc | 74 const TensorShapeProto::Dim& value_dim = value_shape.dim(i); in CombineRaggedTensorToTensorShapes() local
78 if (value_dim.size() >= 0) { in CombineRaggedTensorToTensorShapes()
80 if (output_shape_dim->size() != value_dim.size()) { in CombineRaggedTensorToTensorShapes()
84 "] = ", value_dim.size(), " but shape[", i + ragged_rank, in CombineRaggedTensorToTensorShapes()
88 output_shape_dim->set_size(value_dim.size()); in CombineRaggedTensorToTensorShapes()
114 int value_dim = value_shape.dim(i + 1).size(); in ValidateDefaultValueShape() local
115 if (default_dim >= 0 && value_dim >= 0 && default_dim != 1 && in ValidateDefaultValueShape()
116 default_dim != value_dim) { in ValidateDefaultValueShape()
124 i - default_value_shape.dim_size(), "] = ", value_dim); in ValidateDefaultValueShape()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 40 def test_non_masked_attention(self, value_dim, output_shape, output_dims): argument
45 value_dim=value_dim,
278 value_dim=4,
317 value_dim=4,
332 num_heads=3, key_dim=4, value_dim=4)
|
| D | multi_head_attention.py | 220 value_dim=None, argument
236 self._value_dim = value_dim if value_dim else key_dim
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 190 int64 value_dim = value_shape_.dim_size(0); in Find() local
201 for (int64 j = 0; j < value_dim; j++) { in Find()
211 for (int64 j = 0; j < value_dim; j++) { in Find()
224 int64 value_dim = value_shape_.dim_size(0); in DoInsert() local
232 for (int64 j = 0; j < value_dim; j++) { in DoInsert()
265 int64 value_dim = value_shape_.dim_size(0); in ExportValues() local
272 "values", TensorShape({size, value_dim}), &values)); in ExportValues()
281 for (int64 j = 0; j < value_dim; j++) { in ExportValues()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-multi-head-attention.pbtxt | 132 …argspec: "args=[\'self\', \'num_heads\', \'key_dim\', \'value_dim\', \'dropout\', \'use_bias\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-multi-head-attention.pbtxt | 132 …argspec: "args=[\'self\', \'num_heads\', \'key_dim\', \'value_dim\', \'dropout\', \'use_bias\', \'…
|