Home
last modified time | relevance | path

Searched refs:tensor_shape_proto (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
DBUILD1410 ":tensor_shape_proto",
1434 ":tensor_shape_proto",
1450 ":tensor_shape_proto",
1476 ":tensor_shape_proto",
1494 ":tensor_shape_proto",
1507 ":tensor_shape_proto",
1528 ":tensor_shape_proto",
1543 ":tensor_shape_proto",
1556 ":tensor_shape_proto",
1581 ":tensor_shape_proto",
[all …]
Dtensor_util.h279 TensorShapeProto tensor_shape_proto;
280 internal::SetTensorProtoShape(shape, &tensor_shape_proto);
281 if (TensorShape(tensor_shape_proto).num_elements() != values.size()) {
288 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc1305 const TensorShapeProto& tensor_shape_proto) { in ExpectTensorShape() argument
1307 TensorShape tensor_shape(tensor_shape_proto); in ExpectTensorShape()
1314 TensorShapeProto tensor_shape_proto; in TEST_F() local
1320 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1327 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1332 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1340 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1341 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1349 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1350 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
[all …]
Dop_level_cost_estimator.h31 TensorShapeProto* tensor_shape_proto);
DBUILD34 "//tensorflow/core/framework:tensor_shape_proto",
Dop_level_cost_estimator.cc1374 TensorShapeProto* tensor_shape_proto) { in GetTensorShapeProtoFromTensorProto() argument
1375 tensor_shape_proto->Clear(); in GetTensorShapeProtoFromTensorProto()
1400 tensor_shape_proto->add_dim()->set_size(value); \ in GetTensorShapeProtoFromTensorProto()
Dgraph_properties.cc479 const auto& tensor_shape_proto = tensor_proto.tensor_shape(); in NumElementsFromTensorProto() local
480 if (tensor_shape_proto.unknown_rank()) { in NumElementsFromTensorProto()
484 for (const auto& dim : tensor_shape_proto.dim()) { in NumElementsFromTensorProto()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc257 TensorShapeProto tensor_shape_proto; in ConvertTypeToTensorShape() local
258 ConvertToTensorShapeProto(tensor_type.getShape(), &tensor_shape_proto); in ConvertTypeToTensorShape()
259 return PartialTensorShape(tensor_shape_proto); in ConvertTypeToTensorShape()
/external/tensorflow/tensorflow/core/example/
DBUILD109 "//tensorflow/core/framework:tensor_shape_proto",
/external/tensorflow/tensorflow/core/util/
DBUILD890 "//tensorflow/core/framework:tensor_shape_proto",
904 "//tensorflow/core/framework:tensor_shape_proto",
/external/tensorflow/tensorflow/python/framework/
Dops_test.py3678 tensor_shape_proto = tensor_shape_pb2.TensorShapeProto(dim=[
3682 list_proto = attr_value_pb2.AttrValue.ListValue(shape=[tensor_shape_proto])