Home
last modified time | relevance | path

Searched refs:MutableAttrs (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc45 MutableAttrs()->Set(attr_name, value); in SetAttrValue()
51 MutableAttrs()->Set(attr_name, StringPiece(data, length)); in SetAttrString()
56 MutableAttrs()->Set(attr_name, static_cast<int64>(value)); in SetAttrInt()
61 MutableAttrs()->Set(attr_name, value); in SetAttrFloat()
66 MutableAttrs()->Set(attr_name, value); in SetAttrBool()
71 MutableAttrs()->Set(attr_name, value); in SetAttrType()
93 MutableAttrs()->Set(attr_name, proto); in SetAttrShape()
105 MutableAttrs()->Set(attr_name, attr_value); in SetAttrFunction()
114 MutableAttrs()->Set(attr_name, attr_value); in SetAttrFunctionName()
121 MutableAttrs()->Set(attr_name, t); in SetAttrTensor()
[all …]
Dmkl_eager_op_rewrite_test.cc90 orig_op->MutableAttrs()->Set("T", T); \
91 orig_op->MutableAttrs()->Set("padding", "VALID"); \
104 orig_op->MutableAttrs()->Set("T", T); \
105 orig_op->MutableAttrs()->Set("padding", "EXPLICIT"); \
118 orig_op->MutableAttrs()->Set("T", T); \
119 orig_op->MutableAttrs()->Set("padding", "EXPLICIT"); \
142 orig_op->MutableAttrs()->Set("T", T); \
156 orig_op->MutableAttrs()->Set("T", T); \
157 orig_op->MutableAttrs()->Set("data_format", "" DATA_FORMAT ""); \
Dmkl_eager_op_rewrite.cc150 const NodeDef& orig_ndef = orig_op->MutableAttrs()->BuildNodeDef(); in SetupNewOp()
154 (*new_mkl_op)->MutableAttrs()->Set(attr.first, attr.second); in SetupNewOp()
158 ->MutableAttrs() in SetupNewOp()
246 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in RewriteConv2D()
254 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in RewriteFusedBatchNormV3()
Dexecute.cc240 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes()
389 Fprint128 cache_key = op->MutableAttrs()->CacheKey(op->DeviceName()); in GetOrCreateKernelAndDevice()
479 op->MutableAttrs()->Set(kXlaMustCompileAttr, true); in GetOrCreateKernelAndDevice()
487 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in GetOrCreateKernelAndDevice()
767 pack_op.MutableAttrs()->Set("N", handle->NumPackedHandles()); in MaybePackInputTensor()
768 pack_op.MutableAttrs()->Set("T", handle->dtype); in MaybePackInputTensor()
962 op->MutableAttrs()->BuildNodeDef(), op->EagerContext().FuncLibDef(), in EagerRemoteExecute()
Deager_operation.h151 AttrBuilder* MutableAttrs() { return &attrs_; } in MutableAttrs() function
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc60 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in CreateUncachedKernelAndDeviceOp()
130 op.MutableAttrs()->Set("tensor_name", wire_id_); in StartSend()
131 op.MutableAttrs()->Set("send_device", send_device_->name()); in StartSend()
132 op.MutableAttrs()->Set( in StartSend()
135 op.MutableAttrs()->Set("recv_device", recv_device_->name()); in StartSend()
136 op.MutableAttrs()->Set("client_terminated", false); in StartSend()
138 op.MutableAttrs()->Set("T", src_->dtype); in StartSend()
283 op.MutableAttrs()->Set("tensor_name", wire_id_); in StartRecv()
284 op.MutableAttrs()->Set("send_device", send_device_->name()); in StartRecv()
285 op.MutableAttrs()->Set( in StartRecv()
[all …]
Deager_service_impl.cc140 eager_op->MutableAttrs()->Set(attr.first, attr.second); in GetEagerOperationAndNumRetvals()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc63 const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); in Run()
66 orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); in Run()
85 new_op->MutableAttrs()->CopyAttributes(orig_op->Attrs()); in Run()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc259 op_->MutableAttrs()->NumInputs(inputs_.Size()); in BuildEagerOp()
261 op_->MutableAttrs()->Set(attr.first, attr.second); in BuildEagerOp()
266 op_->MutableAttrs()->CacheKey(op_->DeviceName()); in BuildEagerOp()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc874 operation->MutableAttrs()->Set(attr_name, attr_value); in TFE_OpSetAttrValueProto()