Home
last modified time | relevance | path

Searched defs:tensor_proto (Results 1 – 25 of 62) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dfast_tensor_util.pyx10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
53 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray): argument
60 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray): argument
67 tensor_proto, np.ndarray[np.uint64_t, ndim=1] nparray): argument
74 tensor_proto, np.ndarray[np.uint8_t, ndim=1] nparray): argument
82 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
[all …]
H A Dtensor_util.py46 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
51 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
63 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
68 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
122 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
125 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values): argument
128 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values): argument
131 def SlowAppendInt64ArrayToTensorProto(tensor_proto, proto_values): argument
134 def SlowAppendQIntArrayToTensorProto(tensor_proto, proto_values): argument
137 def SlowAppendUInt32ArrayToTensorProto(tensor_proto, proto_values): argument
[all …]
/aosp_15_r20/external/federated-compute/fcp/aggregation/core/
H A Dtensor_test.cc127 TensorProto tensor_proto; in TEST() local
139 TensorProto tensor_proto; in TEST() local
162 TensorProto tensor_proto; in TEST() local
175 TensorProto tensor_proto; in TEST() local
183 TensorProto tensor_proto; in TEST() local
H A Dtensor.cc215 StatusOr<Tensor> Tensor::FromProto(const TensorProto& tensor_proto) { in FromProto()
228 StatusOr<Tensor> Tensor::FromProto(TensorProto&& tensor_proto) { in FromProto()
243 TensorProto tensor_proto; in ToProto() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor_util_test.cc463 TensorProto tensor_proto = in TEST() local
485 TensorProto tensor_proto = in TEST() local
551 TensorProto tensor_proto; in CreateAsProtoTensorContent() local
562 TensorProto tensor_proto; in CreateAsProtoField() local
582 TensorProto tensor_proto = in ConstantTailTest() local
635 TensorProto tensor_proto; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dunique_op_test.cc41 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
54 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
115 TensorProto tensor_proto; in GetRandomStringsTensorProto() local
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dconverters_test.cc101 tf::TensorProto tensor_proto = PARSE_TEXT_PROTO(R"pb( in TEST() local
121 tf::TensorProto tensor_proto = PARSE_TEXT_PROTO(R"pb( in TEST() local
135 tf::TensorProto tensor_proto = PARSE_TEXT_PROTO(R"pb( in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op_test.cc249 TensorProto tensor_proto; in TEST_F() local
305 TensorProto tensor_proto; in TEST_F() local
356 TensorProto tensor_proto; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc402 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
422 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto()
475 uint64 NumElementsFromTensorProto(const TensorProto& tensor_proto) { in NumElementsFromTensorProto()
1033 const auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
1117 const TensorProto* tensor_proto = ctx->input_tensor_protos[dst_input]; in UpdateNode() local
1625 TensorProto tensor_proto; in UpdateOutputShapesAndValues() local
1714 const TensorProto& tensor_proto = node.attr().at("value").tensor(); in MaybeUpdateNodeContextOutput() local
1967 TensorProto tensor_proto; in MakeIntegerScalarTensorProto() local
1980 const TensorProto& tensor_proto, in MaybeTensorProtoToShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dconvert_tensor.cc62 TensorProto tensor_proto; in ConvertToProto() local
482 TensorProto tensor_proto; in ConvertToTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.cc59 TensorProto tensor_proto; in ConvertToProto() local
481 TensorProto tensor_proto; in ConvertToTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc505 const TensorProto& tensor_proto, in MakeTensorFromProto()
532 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp884 auto tensor_proto = graph_proto_copy.add_initializer(); in GetGraphProtoSize() local
1062 size_t& num_external_data) -> std::string { in AddAttribute()
1280 onnx::TensorProto* tensor_proto, in EncodeTensor()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dtransform_utils_test.cc186 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttr() local
202 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttrWithTensor() local
212 TensorProto tensor_proto; in TestGetNodeTensorAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsingle_threaded_cpu_device.cc61 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
H A Drenamed_device.h107 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Devaluation_utils.cc50 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dvirtual_device.cc79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc66 const tensorflow::TensorProto &tensor_proto, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_grpc_testlib.cc54 const TensorProto& tensor_proto = val.tensor(); in SendEvents() local
H A Ddebug_io_utils.cc129 TensorProto* tensor_proto, in WrapStringTensorAsEvents()
190 TensorProto tensor_proto; in WrapTensorAsEvents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compilation_device.cc147 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dconvert_tf_quant_ops_to_mhlo.cc131 tensorflow::TensorProto tensor_proto; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dsmall_constant_optimization.cc104 TensorProto tensor_proto; in ExtractSmallTensorValue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc802 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
930 tensorflow::TensorProto tensor_proto; in ConstTensorProto() local

123