| /external/tensorflow/tensorflow/core/ops/ |
| D | function_ops.cc | 44 const TensorShapeProto& shape_proto = shape_attr->list().shape(0); in __anonfc9946590102() local
78 const TensorShapeProto& shape_proto = shape_attr->list().shape(0); in __anonfc9946590102() local
|
| D | math_ops_test.cc | 240 auto shape_proto = [](std::initializer_list<int64> dim_sizes) { in TEST() local
|
| D | array_ops_test.cc | 172 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in TEST() local
1847 TensorShapeProto shape_proto; in TEST() local
|
| /external/tensorflow/tensorflow/core/util/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 24 Shape::Shape(const ShapeProto& shape_proto) { in Shape()
197 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 49 TensorShapeProto shape_proto; in TEST() local
|
| D | shape_inference.cc | 115 TensorShapeProto shape_proto; in PropagateShapes() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| 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
|
| D | fold_constants_test.cc | 146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 138 TensorShapeProto shape_proto; in UpdatePlaceholderShape() local
365 TensorShapeProto shape_proto; in GrapplerItemFromMetaGraphDef() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 184 TensorShapeProto shape_proto; in ReplaceBroadcast() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_ops.py | 59 def _create_handle_data_proto(shape_proto, dtype_enum): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util.cc | 186 TensorShapeProto* shape_proto) { in SetTensorProtoShape()
|
| D | function_test.cc | 1508 TensorShapeProto* shape_proto = shape_attr.mutable_list()->add_shape(); in TEST() local
|
| D | op_kernel_test.cc | 810 TensorShapeProto shape_proto; member in tensorflow::__anon843475bc0111::GetAttrKernel
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 454 for (const xla::ShapeProto& shape_proto : executable.input_shapes()) { in TPUExecute() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 353 TensorShapeProto shape_proto; in InferOutputShapes() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 154 TensorShapeProto shape_proto; in CalculateOutputSize() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer_test.cc | 271 TensorShapeProto shape_proto; in SetShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 255 tensorflow::TensorShapeProto shape_proto; in SetMetadataProtoArgs() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 140 string PrintTensorShape(const TensorShapeProto& shape_proto) { in PrintTensorShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 190 xla::ShapeProto shape_proto; in DecodeSelfDescribingShapeConstant() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 807 for (const ShapeProto& shape_proto : arg->input_shape_with_layout()) { in Compile() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 383 TensorShapeProto shape_proto; in SetArgShape() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 400 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in MakeTensorProtoFromShape() local
|