Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/util/tensor_bundle/
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()
/external/tensorflow/tensorflow/core/kernels/
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()
/external/tensorflow/tensorflow/core/distributed_runtime/
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 …]
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()
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()
Dcluster_function_library_runtime.cc283 args[i].AsProtoTensorContent(send->mutable_tensor()); in Run()
310 mapped_recvs[recv.name()] = recv.mutable_tensor(); in Run()
/external/tensorflow/tensorflow/core/framework/
Dlog_memory.cc57 tensor.FillDescription(allocation.mutable_tensor()); in RecordTensorAllocation()
76 tensor.FillDescription(output.mutable_tensor()); in RecordTensorOutput()
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()
Dattr_value_util_test.cc200 (*b.mutable_tensor()->mutable_string_val(0))[3] = '1'; in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
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()
Dnode_file_writer.cc154 input.AsProtoField(input_tensor.mutable_tensor()); in RecordNodeExecution()
/external/tensorflow/tensorflow/core/util/
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()
Dtensor_slice_writer.h138 SavedSliceMeta* ssm = sts_.mutable_meta()->mutable_tensor(index); in Add()
/external/tensorflow/tensorflow/core/grappler/utils/
Dcanonicalizer.cc62 tensor::CompressTensorProtoInPlace(attr_val.mutable_tensor()); in CompressConstants()
/external/tensorflow/tensorflow/core/summary/
Dsummary_file_writer.cc92 t.AsProtoField(v->mutable_tensor()); in WriteTensor()
94 t.AsProtoTensorContent(v->mutable_tensor()); in WriteTensor()
/external/tensorflow/tensorflow/python/client/
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()
/external/tensorflow/tensorflow/core/runtime_fallback/util/
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()
/external/tensorflow/tensorflow/core/debug/
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()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
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()
/external/tensorflow/tensorflow/lite/toco/
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 …]
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Droundtrip.cc39 auto* tensor = attr_val.mutable_tensor(); in NormalizeNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc178 TensorProto* t = (*node.mutable_attr())["value"].mutable_tensor(); in Optimize()
/external/tensorflow/tensorflow/core/data/service/
Dtest_util.cc191 .mutable_tensor()); in InterleaveTextlineDataset()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.cc159 val.AsProtoTensorContent(response.mutable_tensor()); in EncodeTensorToByteBuffer()
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model.cc163 (*const_node->mutable_attr())["value"].mutable_tensor()); in ConvertVariableToConstant()

123