Home
last modified time | relevance | path

Searched refs:InitAlloc (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.h47 void InitAlloc(DeviceBase* d, const AllocatorAttributes& aa);
Dtensor_coding_test.cc93 response.InitAlloc(&cpu_device, AllocatorAttributes()); in Validate()
184 response.InitAlloc(&cpu_device, AllocatorAttributes()); in BM_TensorResponse()
Dtensor_coding.cc42 void TensorResponse::InitAlloc(DeviceBase* d, const AllocatorAttributes& aa) { in InitAlloc() function in tensorflow::TensorResponse
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr.cc139 resp_.InitAlloc(dst_device_, alloc_attrs_); in StartRTCall()