Home
last modified time | relevance | path

Searched refs:mutable_tensor_shape (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dunique_op_test.cc43 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 Dvariant_op_copy_test.cc178 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 Dtensor_util.h288 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_test.cc405 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 Dtrt_convert_api_test.cc169 (*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 Dexport_tensorflow.cc114 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 Dimport_tensorflow_test.cc119 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 Dconvert_tensor.cc216 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 Dconvert_tensor.cc216 auto* shape = tensor_copy.mutable_tensor_shape(); in ConvertTensorProto()
400 ConvertToTensorShapeProto(shape, output->mutable_tensor_shape()); in ConvertToTensorProto()
H A Dconvert_tensor_test.cc187 tensor.mutable_tensor_shape()->add_dim()->set_size(1ULL << 35); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dconstant_utils.cc56 tensorflow::TensorShapeProto* shape = repr.mutable_tensor_shape(); in CreateConstOpWithSingleValue()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dquantization_ops_test.cc327 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 Dsmall_constant_optimization.cc147 shape.AsProto(tensor_proto.mutable_tensor_shape()); in ExtractSmallTensorValue()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer.cc180 *t->mutable_tensor_shape() = TensorShapeProto(); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcompression_utils.cc57 component.shape().AsProto(metadata->mutable_tensor_shape()); in CompressElement()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_tensor_coding.cc124 val.shape().AsProto(skeleton.mutable_tensor_shape()); in EncodeSkeleton()
/aosp_15_r20/external/tensorflow/tensorflow/cc/ops/
H A Dconst_op_test.cc107 TensorShape({2, 2}).AsProto(proto.mutable_tensor_shape()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dtensor_coding.cc174 !ReadNestedMessage(input, tensor_meta->mutable_tensor_shape())) in ParseTensorSubmessage()
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_rewrite.cc110 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 Dresolve_svdf.cc187 allocated_tensor->mutable_tensor_shape(); in MaybeMergeConstNodes()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcontrol_flow_deps_to_chains.cc158 att_value.mutable_tensor()->mutable_tensor_shape(); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dshape_inference.cc128 value.mutable_tensor_shape()->add_dim()->set_size( in PropagateShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc359 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 Dop_level_cost_estimator_test.cc404 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 Dgpu_util.cc141 tensor.shape().AsProto(proto->mutable_tensor_shape()); in SetProtoFromGPU()

12