Home
last modified time | relevance | path

Searched defs:alloc_attrs (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstack.cc47 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
H A Dpad_op.cc110 AllocatorAttributes alloc_attrs; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_test.cc1136 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_util.cc24 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensorsToRendezvous()
58 const std::vector<AllocatorAttributes>& alloc_attrs, in RecvOutputsFromRendezvousAsync()
H A Dsingle_threaded_cpu_device.cc62 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
H A Drenamed_device.h108 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
H A Dthreadpool_device.cc122 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
H A Dprocess_function_library_runtime.cc124 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensors()
143 const std::vector<AllocatorAttributes>& alloc_attrs, in ReceiveTensorsAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device.cc321 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToPluggableDevice()
363 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc506 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
533 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgpu_swapping_kernels.cc37 AllocatorAttributes alloc_attrs; in ComputeAsync() local
H A Devaluation_utils.cc51 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dvirtual_device.cc80 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc67 const tensorflow::AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compilation_device.cc147 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drendezvous.h53 AllocatorAttributes alloc_attrs; member
H A Ddevice_base.h251 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
H A Dop_kernel_test.cc456 AllocatorAttributes alloc_attrs; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dsummary_op_test.cc77 AllocatorAttributes alloc_attrs; in TestScalarSummaryOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Drpc_rendezvous_mgr.cc61 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
H A Dgrpc_worker_service.cc535 AllocatorAttributes alloc_attrs; in GrpcRecvTensorAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/
H A Draw_api_test.cc393 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc773 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to, in MaybeCopyTensorToGPU()
825 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc803 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()