Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/framework/kernels/
Dzero_initializer_op.cc48 attr.set_nic_compatible(true); in Compute()
107 attr.set_nic_compatible(true); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc54 aa.set_nic_compatible(nic_compatible); in TEST()
77 a.set_nic_compatible(true); in TEST()
Dallocator.h362 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.h59 attr.set_nic_compatible(true); in EnsureSparseVariableAccess()
220 attr.set_nic_compatible(true); in PrepareToUpdateVariable()
Dassign_op.h60 attr.set_nic_compatible(true); in Compute()
Dcount_up_to_op.cc86 attr.set_nic_compatible(true); in Compute()
Ddense_update_functor.cc90 attr.set_nic_compatible(true); \
Dresource_variable_ops.cc367 attr.set_nic_compatible(true); in Compute()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_memory_manager.cc377 alloc_attrs.set_nic_compatible(true); in TransportOptionsFromTensor()
460 alloc_attrs.set_nic_compatible(true); in TensorFromTransportOptions()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc592 cpu_attr.set_nic_compatible(true); in RecvBufAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc800 attr->set_nic_compatible(true); in InferAllocAttr()
825 attr->set_nic_compatible(true); in InferAllocAttr()
843 attr->set_nic_compatible(true); in InferAllocAttr()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1149 host_alloc_attrs.set_nic_compatible(true); in Clone()