Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc75 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo() local
77 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()
90 TensorShapeProto& tensor_shape_proto = in BuildRemoteFusedGraphExecuteInfo() local
93 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()
Dhexagon_graph_execution_test.cc220 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local
222 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
232 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local
234 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
/external/tensorflow/tensorflow/core/framework/
DBUILD1392 ":tensor_shape_proto",
1416 ":tensor_shape_proto",
1431 ":tensor_shape_proto",
1457 ":tensor_shape_proto",
1475 ":tensor_shape_proto",
1488 ":tensor_shape_proto",
1509 ":tensor_shape_proto",
1523 ":tensor_shape_proto",
1536 ":tensor_shape_proto",
1554 ":tensor_shape_proto",
[all …]
Dtensor_util.h282 TensorShapeProto tensor_shape_proto;
283 internal::SetTensorProtoShape(shape, &tensor_shape_proto);
284 if (TensorShape(tensor_shape_proto).num_elements() != values.size()) {
291 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc193 TensorShapeProto tensor_shape_proto; in ConvertTypeToTensorShape() local
194 ConvertToTensorShapeProto(tensor_type.getShape(), &tensor_shape_proto); in ConvertTypeToTensorShape()
195 return PartialTensorShape(tensor_shape_proto); in ConvertTypeToTensorShape()
/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.cc1363 TensorShapeProto* tensor_shape_proto) { in GetTensorShapeProtoFromTensorProto() argument
1364 tensor_shape_proto->Clear(); in GetTensorShapeProtoFromTensorProto()
1389 tensor_shape_proto->add_dim()->set_size(value); \ in GetTensorShapeProtoFromTensorProto()
/external/tensorflow/tensorflow/core/example/
DBUILD108 "//tensorflow/core/framework:tensor_shape_proto",
/external/tensorflow/tensorflow/core/util/
DBUILD841 "//tensorflow/core/framework:tensor_shape_proto",
855 "//tensorflow/core/framework:tensor_shape_proto",
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc633 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo() local
635 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()
658 TensorShapeProto& tensor_shape_proto = in BuildRemoteFusedGraphExecuteInfo() local
661 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()