Searched defs:value_dim (Results 1 – 5 of 5) sorted by relevance
74 const TensorShapeProto::Dim& value_dim = value_shape.dim(i); in CombineRaggedTensorToTensorShapes() local114 int value_dim = value_shape.dim(i + 1).size(); in ValidateDefaultValueShape() local
36 def test_non_masked_attention(self, value_dim, output_shape, output_dims): argument
220 value_dim=None, argument
239 int64_t value_dim = value_shape_.dim_size(0); in Find() local273 int64_t value_dim = value_shape_.dim_size(0); in DoInsert() local314 int64_t value_dim = value_shape_.dim_size(0); in ExportValues() local391 int64_t value_dim = value_shape_.dim_size(0); in ExportKeysAndValues() local
1621 Value value_dim = to_shape_scalar_type( in reifyReturnTypeShapes() local2458 Value value_dim = to_shape_scalar_type( in reifyReturnTypeShapes() local2835 Value value_dim = in reifyReturnTypeShapes() local3755 Value value_dim = to_shape_scalar_type( in reifyReturnTypeShapes() local