Home
last modified time | relevance | path

Searched refs:mutable_tensor (Results 1 – 25 of 59) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dbyte_swap.cc158 auto tsize = node_value.mutable_tensor()->tensor_content().size(); in ByteSwapTensorContent()
159 auto p_type = node_value.mutable_tensor()->dtype(); in ByteSwapTensorContent()
163 DCHECK(parsed.FromProto(*node_value.mutable_tensor())); in ByteSwapTensorContent()
167 .mutable_tensor() in ByteSwapTensorContent()
175 string(node_value.mutable_tensor()->tensor_content()) in ByteSwapTensorContent()
181 .mutable_tensor() in ByteSwapTensorContent()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmessage_wrappers.cc119 feeds_[i].second.AsProtoTensorContent(feed->mutable_tensor()); in ToProto()
171 TensorProto* value_proto = feed->mutable_tensor(); in add_feed()
412 sends_[i].second.AsProtoTensorContent(send->mutable_tensor()); in ToProto()
489 TF_RETURN_IF_ERROR(run_step_request.FeedValue(i, send->mutable_tensor())); in AddSendFromRunStepRequest()
498 *send->mutable_tensor() = run_callable_request.feed(i); in AddSendFromRunCallableRequest()
683 out_tensor->Swap(response_.mutable_recv(i)->mutable_tensor()); in RecvValue()
699 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
758 out_tensor->Swap(response_->mutable_recv(i)->mutable_tensor()); in RecvValue()
774 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
894 return run_graph_response->RecvValue(i, response_tensor->mutable_tensor()); in AddTensorFromRunGraphResponse()
[all …]
H A Dtensor_coding.cc65 meta_.mutable_tensor()->Swap(&empty); in InitFrom()
95 meta_.mutable_tensor()->Swap(&empty); in ParseFrom()
236 !ParseTensorSubmessage(&input, meta_.mutable_tensor())) { in ParseFast()
292 meta_.mutable_tensor()->Swap(&empty); in ParseSlow()
H A Dtensor_coding_test.cc82 src.AsProtoTensorContent(proto.mutable_tensor()); in Validate()
84 src.AsProtoField(proto.mutable_tensor()); in Validate()
170 src.AsProtoTensorContent(proto.mutable_tensor()); in MakeFloatTensorTestCase()
H A Dcluster_function_library_runtime.cc283 args[i].AsProtoTensorContent(send->mutable_tensor()); in Run()
310 mapped_recvs[recv.name()] = recv.mutable_tensor(); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsummary_tensor_op.cc50 tensor.AsProtoField(v->mutable_tensor()); in Compute()
52 tensor.AsProtoTensorContent(v->mutable_tensor()); in Compute()
92 tensor.AsProtoField(v->mutable_tensor()); in Compute()
94 tensor.AsProtoTensorContent(v->mutable_tensor()); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcontrol_flow_deps_to_chains.cc157 att_value.mutable_tensor()->set_dtype(DT_FLOAT); in Run()
158 att_value.mutable_tensor()->mutable_tensor_shape(); in Run()
159 att_value.mutable_tensor()->add_int_val(0); in Run()
H A Dnode_file_writer.cc154 input.AsProtoField(input_tensor.mutable_tensor()); in RecordNodeExecution()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_reader_test.cc369 for (auto& tensor : *sts.mutable_meta()->mutable_tensor()) { in TEST()
397 for (auto& tensor : *sts.mutable_meta()->mutable_tensor()) { in TEST()
425 for (auto& tensor : *sts.mutable_meta()->mutable_tensor()) { in TEST()
450 for (auto& tensor : *sts.mutable_meta()->mutable_tensor()) { in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dlog_memory.cc57 tensor.FillDescription(allocation.mutable_tensor()); in RecordTensorAllocation()
76 tensor.FillDescription(output.mutable_tensor()); in RecordTensorOutput()
H A Dattr_value_util.cc566 value.AsProtoTensorContent(out->mutable_tensor()); in SetAttrValue()
568 value.AsProtoField(out->mutable_tensor()); in SetAttrValue()
584 *out->mutable_tensor() = value; in SetAttrValue()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dattr_util.cc246 auto* mutable_tensor = attr_tmp.mutable_tensor(); in FillAttrValueMapUsingScalar() local
248 tf_tensor->AsProtoTensorContent(mutable_tensor); in FillAttrValueMapUsingScalar()
250 tf_tensor->AsProtoField(mutable_tensor); in FillAttrValueMapUsingScalar()
558 tf_tensor.AsProtoTensorContent(tf_attr->mutable_tensor()); in SetUpScalarAttr()
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession_ref.cc144 it.second.AsProtoField(feed->mutable_tensor()); in RecordRun()
171 tensor.AsProtoField(tproto->mutable_tensor()); in RecordRun()
292 input.second.AsProtoField(feed->mutable_tensor()); in RecordPRun()
304 tensor.AsProtoField(tproto->mutable_tensor()); in RecordPRun()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc171 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertFloatTensorConst()
213 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertFloatTensorConst()
236 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertBoolTensorConst()
260 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertIntTensorConst()
284 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in CreateIntTensorConst()
309 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertComplex64TensorConst()
332 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in CreateMatrixShapeTensorConst()
350 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in CreateDummyConcatDimTensorConst()
365 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in CreateReshapeShapeTensorConst()
755 (*max_bounds_const_op->mutable_attr())["value"].mutable_tensor(); in ConvertRelu1Operator()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_io_utils.cc105 value->mutable_tensor()->set_dtype(tensor_dtype); in PrepareChunkEventProto()
106 *value->mutable_tensor()->mutable_tensor_shape() = tensor_shape; in PrepareChunkEventProto()
165 value->mutable_tensor()->mutable_string_val()->Swap( in WrapStringTensorAsEvents()
171 value->mutable_tensor()->add_string_val(strs[j]); in WrapStringTensorAsEvents()
218 ->mutable_tensor() in WrapTensorAsEvents()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc200 *operation->add_op_inputs()->mutable_tensor() = in BuildOperation()
417 SetTensorProto(val.mutable_tensor()); in TEST_F()
520 SetTensorProto(val.mutable_tensor()); in TestFunction()
608 SetTensorProto(val.mutable_tensor()); in TestComponentFunction()
774 SetTensorProto(val.mutable_tensor()); in Init()
1169 SetTensorProto(lcoal_handle->mutable_tensor()); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/summary/
H A Dsummary_file_writer.cc92 t.AsProtoField(v->mutable_tensor()); in WriteTensor()
94 t.AsProtoTensorContent(v->mutable_tensor()); in WriteTensor()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dnamed_tensor.pb.h190 ::tensorflow::TensorProto* mutable_tensor(); in Swap()
325 inline ::tensorflow::TensorProto* NamedTensorProto::mutable_tensor() { in mutable_tensor() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dattr_value.pb.h322 ::tensorflow::TensorProto* mutable_tensor(int index); in Swap()
324 mutable_tensor(); in Swap()
607 ::tensorflow::TensorProto* mutable_tensor(); in Swap()
1216 inline ::tensorflow::TensorProto* AttrValue_ListValue::mutable_tensor(int index) { in mutable_tensor() function
1221 AttrValue_ListValue::mutable_tensor() { in mutable_tensor() function
1645 inline ::tensorflow::TensorProto* AttrValue::mutable_tensor() { in mutable_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dcanonicalizer.cc62 tensor::CompressTensorProtoInPlace(attr_val.mutable_tensor()); in CompressConstants()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/util/
Dsaved_tensor_slice.pb.h380 ::tensorflow::SavedSliceMeta* mutable_tensor(int index); in Swap()
382 mutable_tensor(); in Swap()
974 inline ::tensorflow::SavedSliceMeta* SavedTensorSliceMeta::mutable_tensor(int index) { in mutable_tensor() function
979 SavedTensorSliceMeta::mutable_tensor() { in mutable_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Droundtrip.cc39 auto* tensor = attr_val.mutable_tensor(); in NormalizeNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer.cc178 TensorProto* t = (*node.mutable_attr())["value"].mutable_tensor(); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dsingle_machine_test.cc264 (*shp->mutable_attr())["value"].mutable_tensor()); in RunInfiniteTFLoop()
296 (*t->mutable_attr())["value"].mutable_tensor()); in RunInfiniteTFLoop()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dtest_util.cc191 .mutable_tensor()); in InterleaveTextlineDataset()

123