| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | ragged_to_dense_util_test.cc | 30 TensorShapeProto shape_proto; in TEST() local
44 TensorShapeProto shape_proto; in TEST() local
60 TensorShapeProto shape_proto; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | variable_ops.cc | 34 TensorShapeProto shape_proto; member
166 StatusOr<TensorShapeProto> shape_proto = in ValidateImpl() local
270 StatusOr<TensorShapeProto> shape_proto = in ValidateImpl() local
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/core/ |
| H A D | tensor_shape.cc | 41 const TensorShapeProto& shape_proto) { in FromProto()
55 TensorShapeProto shape_proto; in ToProto() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | function_ops.cc | 44 const TensorShapeProto& shape_proto = shape_attr->list().shape(0); in __anon748b2ccf0102() local
78 const TensorShapeProto& shape_proto = shape_attr->list().shape(0); in __anon748b2ccf0102() local
|
| H A D | array_ops_test.cc | 172 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in TEST() local
1860 TensorShapeProto shape_proto; in TEST() local
|
| H A D | math_ops_test.cc | 240 auto shape_proto = [](std::initializer_list<int64_t> dim_sizes) { in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape.cc | 24 Shape::Shape(const ShapeProto& shape_proto) { in Shape()
223 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | summarize_graph_main.cc | 46 TensorShapeProto shape_proto = node->attr().at("shape").shape(); in PrintNodeInfo() local
83 TensorShapeProto shape_proto = node->attr().at("shape").shape(); in PrintBenchmarkUsage() local
|
| H A D | fold_constants_test.cc | 146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | grappler_item_builder.cc | 138 TensorShapeProto shape_proto; in UpdatePlaceholderShape() local
366 TensorShapeProto shape_proto; in GrapplerItemFromMetaGraphDef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_util_test.cc | 49 TensorShapeProto shape_proto; in TEST() local
|
| H A D | shape_inference.cc | 115 TensorShapeProto shape_proto; in PropagateShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | node_file_writer.cc | 145 TensorShapeProto* shape_proto = input_shapes.mutable_list()->add_shape(); in RecordNodeExecution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| H A D | trt_convert_api_test.cc | 164 TensorShapeProto shape_proto; in GetModel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_rewrite.cc | 184 TensorShapeProto shape_proto; in ReplaceBroadcast() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| H A D | sparse_csr_matrix_ops.py | 53 def _create_handle_data_proto(shape_proto, dtype_enum): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 458 for (const xla::ShapeProto& shape_proto : executable.input_shapes()) { in TPUExecute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | host_compute_ops.cc | 374 TensorShapeProto shape_proto; in InferOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_util.cc | 186 TensorShapeProto* shape_proto) { in SetTensorProtoShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_tensor_op.cc | 154 TensorShapeProto shape_proto; in CalculateOutputSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | scoped_allocator_optimizer_test.cc | 271 TensorShapeProto shape_proto; in SetShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_rewrite_pass.cc | 225 tensorflow::TensorShapeProto shape_proto; in SetMetadataProtoArgs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | cc_op_gen.cc | 140 string PrintTensorShape(const TensorShapeProto& shape_proto) { in PrintTensorShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 870 for (const ShapeProto& shape_proto : arg->input_shape_with_layout()) { in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_runtime.cc | 238 xla::ShapeProto shape_proto; in DecodeSelfDescribingShapeConstant() local
|