Home
last modified time | relevance | path

Searched defs:tensor_shape_proto (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A D_proto_graph.py24 def tensor_shape_proto(outputsize): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dconvert_tensor.cc250 TensorShapeProto tensor_shape_proto; in ConvertTypeToTensorShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.cc252 TensorShapeProto tensor_shape_proto; in ConvertTypeToTensorShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc1339 const TensorShapeProto& tensor_shape_proto) { in ExpectTensorShape()
1348 TensorShapeProto tensor_shape_proto; in TEST_F() local
H A Dop_level_cost_estimator.cc1375 TensorShapeProto* tensor_shape_proto) { in GetTensorShapeProtoFromTensorProto()
H A Dgraph_properties.cc479 const auto& tensor_shape_proto = tensor_proto.tensor_shape(); in NumElementsFromTensorProto() local