/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | unique_op_test.cc | 43 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim); in GetRandomInt32TensorProto() 44 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false); in GetRandomInt32TensorProto() 56 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim); in GetRandomInt32TensorProtoWithRepeat() 57 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false); in GetRandomInt32TensorProtoWithRepeat() 117 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim); in GetRandomStringsTensorProto() 118 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false); in GetRandomStringsTensorProto()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | variant_op_copy_test.cc | 178 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST() 212 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST() 255 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
|
H A D | tensor_util.h | 288 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
|
/aosp_15_r20/external/tensorflow/tensorflow/c/ |
H A D | kernels_test.cc | 405 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2); in TEST_F() 406 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(3); in TEST_F() 458 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2); in TEST_F() 459 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2); in TEST_F() 470 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2); in TEST_F() 471 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(3); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
H A D | trt_convert_api_test.cc | 169 (*signature_def.mutable_inputs())["input"].mutable_tensor_shape()->CopyFrom( in GetModel() 174 .mutable_tensor_shape() in GetModel()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
H A D | export_tensorflow.cc | 114 auto* shape = output_tensor->mutable_tensor_shape(); in ExportFloatArray() 243 auto* shape = tensor->mutable_tensor_shape(); in ConvertBoolTensorConst() 267 auto* shape = tensor->mutable_tensor_shape(); in ConvertIntTensorConst() 289 auto* tensor_shape = tensor->mutable_tensor_shape(); in CreateIntTensorConst() 317 auto* shape = tensor->mutable_tensor_shape(); in ConvertComplex64TensorConst() 337 auto* shape = tensor->mutable_tensor_shape(); in CreateMatrixShapeTensorConst() 374 auto* tensor_shape = tensor->mutable_tensor_shape(); in CreateReshapeShapeTensorConst() 916 sum_reduction_indices_tensor->mutable_tensor_shape(); in ConvertL2NormalizationOperator() 1711 auto* shape = tensor->mutable_tensor_shape(); in ConvertPadOperator() 1743 auto* shape = tensor->mutable_tensor_shape(); in ConvertPadV2Operator() [all …]
|
H A D | import_tensorflow_test.cc | 119 auto* s = t.mutable_tensor_shape(); in BuildConstNode() 474 auto* s = t.mutable_tensor_shape(); in NodeWithTensorContent()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
H A D | convert_tensor.cc | 216 auto* shape = tensor_copy.mutable_tensor_shape(); in ConvertTensorProto() 399 ConvertToTensorShapeProto(shape, output->mutable_tensor_shape()); in ConvertToTensorProto()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
H A D | convert_tensor.cc | 216 auto* shape = tensor_copy.mutable_tensor_shape(); in ConvertTensorProto() 400 ConvertToTensorShapeProto(shape, output->mutable_tensor_shape()); in ConvertToTensorProto()
|
H A D | convert_tensor_test.cc | 187 tensor.mutable_tensor_shape()->add_dim()->set_size(1ULL << 35); in TEST()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
H A D | constant_utils.cc | 56 tensorflow::TensorShapeProto* shape = repr.mutable_tensor_shape(); in CreateConstOpWithSingleValue()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
H A D | quantization_ops_test.cc | 327 input_shape.AsProto(input.mutable_tensor_shape()); in GetModel() 339 output_shape.AsProto(output.mutable_tensor_shape()); in GetModel()
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
H A D | small_constant_optimization.cc | 147 shape.AsProto(tensor_proto.mutable_tensor_shape()); in ExtractSmallTensorValue()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | shape_optimizer.cc | 180 *t->mutable_tensor_shape() = TensorShapeProto(); in Optimize()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
H A D | compression_utils.cc | 57 component.shape().AsProto(metadata->mutable_tensor_shape()); in CompressElement()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
H A D | grpc_tensor_coding.cc | 124 val.shape().AsProto(skeleton.mutable_tensor_shape()); in EncodeSkeleton()
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/ops/ |
H A D | const_op_test.cc | 107 TensorShape({2, 2}).AsProto(proto.mutable_tensor_shape()); in TEST()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
H A D | tensor_coding.cc | 174 !ReadNestedMessage(input, tensor_meta->mutable_tensor_shape())) in ParseTensorSubmessage()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
H A D | nccl_rewrite.cc | 110 result.mutable_tensor_shape()->add_dim()->set_size(shape.dim_size()); in TensorFromShape()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
H A D | resolve_svdf.cc | 187 allocated_tensor->mutable_tensor_shape(); in MaybeMergeConstNodes()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | control_flow_deps_to_chains.cc | 158 att_value.mutable_tensor()->mutable_tensor_shape(); in Run()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
H A D | shape_inference.cc | 128 value.mutable_tensor_shape()->add_dim()->set_size( in PropagateShapes()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
H A D | flatbuffer_import.cc | 359 tensorflow::TensorShapeProto* shape = ret.mutable_tensor_shape(); in ConvertTfliteConstTensor() 585 repr.mutable_tensor_shape()->add_dim()->set_size(buffer.size()); in BuildSparseConstOp() 588 repr.mutable_tensor_shape()->add_dim()->set_size( in BuildSparseConstOp()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
H A D | op_level_cost_estimator_test.cc | 404 temp_tensor_proto.mutable_tensor_shape()->add_dim()->set_size(x); in GetTensorProto() 1352 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(255); in TEST_F() 1358 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(1); in TEST_F() 1359 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
H A D | gpu_util.cc | 141 tensor.shape().AsProto(proto->mutable_tensor_shape()); in SetProtoFromGPU()
|