| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_util.cc | 24 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensorsToRendezvous()
58 const std::vector<AllocatorAttributes>& alloc_attrs, in RecvOutputsFromRendezvousAsync()
|
| D | single_threaded_cpu_device.cc | 66 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | threadpool_device.cc | 92 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | renamed_device.h | 92 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | process_function_library_runtime.cc | 114 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensors()
133 const std::vector<AllocatorAttributes>& alloc_attrs, Rendezvous* rendezvous, in ReceiveTensorsAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 47 AllocatorAttributes alloc_attrs; member
236 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync() local
263 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync() local
|
| D | pad_op.cc | 117 AllocatorAttributes alloc_attrs; in Compute() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 37 AllocatorAttributes alloc_attrs; in ComputeAsync() local
|
| D | evaluation_utils.cc | 72 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.cc | 46 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 50 AllocatorAttributes alloc_attrs; member
|
| D | device_base.h | 244 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 119 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 118 AllocatorAttributes alloc_attrs; in GrpcRecvTensorAsync() local
|
| D | gdr_memory_manager.cc | 375 AllocatorAttributes alloc_attrs; in TransportOptionsFromTensor() local
458 AllocatorAttributes alloc_attrs; in TensorFromTransportOptions() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 62 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
|
| D | grpc_worker_service.cc | 496 AllocatorAttributes alloc_attrs; in GrpcRecvTensorAsync() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 255 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
285 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
312 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
338 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 441 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 639 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToGPU()
685 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|