Home
last modified time | relevance | path

Searched refs:set_nic_compatible (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc55 aa.set_nic_compatible(nic_compatible); in TEST()
78 a.set_nic_compatible(true); in TEST()
98 b.set_nic_compatible(true); in TEST()
Dallocator.h309 void set_nic_compatible(bool v) { value |= (static_cast<int>(v) << 1); } in set_nic_compatible() function
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h60 attr.set_nic_compatible(true); in EnsureSparseVariableAccess()
219 attr.set_nic_compatible(true); in PrepareToUpdateVariable()
Dassign_op.h60 attr.set_nic_compatible(true); in Compute()
Dcount_up_to_op.cc85 attr.set_nic_compatible(true); in Compute()
Ddense_update_functor.cc90 attr.set_nic_compatible(true); \
Dresource_variable_ops.cc413 attr.set_nic_compatible(true); in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc671 cpu_attr.set_nic_compatible(true); in RecvBufAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc850 attr->set_nic_compatible(true); in InferAllocAttr()
875 attr->set_nic_compatible(true); in InferAllocAttr()
893 attr->set_nic_compatible(true); in InferAllocAttr()