| /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 | 62 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | threadpool_device.cc | 89 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | renamed_device.h | 107 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | process_function_library_runtime.cc | 121 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensors()
140 const std::vector<AllocatorAttributes>& alloc_attrs, in ReceiveTensorsAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 47 AllocatorAttributes alloc_attrs; member
237 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync() local
264 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync() local
|
| D | pad_op.cc | 109 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 | 51 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 476 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
500 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
513 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeFastMemTensorFromProto()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 80 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 147 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 53 AllocatorAttributes alloc_attrs; member
|
| D | device_base.h | 239 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | op_kernel_test.cc | 453 AllocatorAttributes alloc_attrs; in TEST_F() local
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op_test.cc | 77 AllocatorAttributes alloc_attrs; in TestScalarSummaryOp() local
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 816 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
853 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
886 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
959 AllocatorAttributes alloc_attrs; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 63 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
|
| D | grpc_worker_service.cc | 530 AllocatorAttributes alloc_attrs; in GrpcRecvTensorAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 635 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToGPU()
687 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 394 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
490 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
517 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
543 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
|