Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api.cc252 op->operation.MutableAttrs()->Set(type_attr, input->handle->dtype); in OpInferSingleInputAttrs()
264 op->operation.MutableAttrs()->Set(input_def.number_attr(), num_inputs); in OpInferSingleTypeInputListAttrs()
268 op->operation.MutableAttrs()->Set(input_def.type_attr(), in OpInferSingleTypeInputListAttrs()
284 op->operation.MutableAttrs()->Set( in OpInferMixedTypeInputListAttrs()
655 op->operation.MutableAttrs()->Set( in TFE_OpSetAttrString()
661 op->operation.MutableAttrs()->Set(attr_name, static_cast<int64>(value)); in TFE_OpSetAttrInt()
665 op->operation.MutableAttrs()->Set(attr_name, value); in TFE_OpSetAttrFloat()
669 op->operation.MutableAttrs()->Set(attr_name, (value == 0) ? false : true); in TFE_OpSetAttrBool()
673 op->operation.MutableAttrs()->Set(attr_name, in TFE_OpSetAttrType()
696 op->operation.MutableAttrs()->Set(attr_name, proto); in TFE_OpSetAttrShape()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc226 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes()
378 op->MutableAttrs()->CacheKey(maybe_unspecified_device_name); in EagerLocalExecute()
401 op->MutableAttrs()->Set(kXlaCompileAttr, true); in EagerLocalExecute()
406 const NodeDef& ndef = op->MutableAttrs()->BuildNodeDef(); in EagerLocalExecute()
1068 op.MutableAttrs()->Set("tensor_name", wire_id); in ExecuteSend()
1069 op.MutableAttrs()->Set("send_device", device->name()); in ExecuteSend()
1070 op.MutableAttrs()->Set( in ExecuteSend()
1073 op.MutableAttrs()->Set("recv_device", recv_device); in ExecuteSend()
1074 op.MutableAttrs()->Set("client_terminated", false); in ExecuteSend()
1076 op.MutableAttrs()->Set("T", h->dtype); in ExecuteSend()
[all …]
Deager_operation.h44 tensorflow::AttrBuilder* MutableAttrs() { return &attrs_; } in MutableAttrs() function
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc254 op_->MutableAttrs()->NumInputs(inputs_.Size()); in BuildEagerOp()
256 op_->MutableAttrs()->Set(attr.first, attr.second); in BuildEagerOp()
262 op_->MutableAttrs()->CacheKey(device == nullptr ? "unspecified" in BuildEagerOp()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc182 op->MutableAttrs()->Set(attr.first, attr.second); in ExecuteOp()