| /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 | renamed_device.h | 108 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | threadpool_device.cc | 122 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | process_function_library_runtime.cc | 124 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensors()
143 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 | 110 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/core/common_runtime/pluggable_device/ |
| D | pluggable_device.cc | 321 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToPluggableDevice()
363 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 80 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 67 const tensorflow::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/c/ |
| D | kernels_test.cc | 1136 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
1173 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
1206 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
1247 TF_AllocatorAttributes alloc_attrs; in TEST_F() local
1295 AllocatorAttributes alloc_attrs; in TEST_F() local
1367 AllocatorAttributes alloc_attrs; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 53 AllocatorAttributes alloc_attrs; member
|
| D | device_base.h | 251 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | op_kernel_test.cc | 456 AllocatorAttributes alloc_attrs; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 506 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
533 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op_test.cc | 77 AllocatorAttributes alloc_attrs; in TestScalarSummaryOp() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 61 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
|
| D | grpc_worker_service.cc | 535 AllocatorAttributes alloc_attrs; in GrpcRecvTensorAsync() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 393 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
489 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
516 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
542 ops::XRTAllocateFromTensor::Attrs alloc_attrs = in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 773 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToGPU()
825 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 803 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|