Home
last modified time | relevance | path

Searched refs:AllocatorAttributes (Results 1 – 25 of 147) sorted by relevance

123456

/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc52 AllocatorAttributes aa; in TEST()
65 AllocatorAttributes a, b; in TEST()
157 AllocatorAttributes DeviceAllocatorAttribute() { in DeviceAllocatorAttribute()
158 AllocatorAttributes attr; in DeviceAllocatorAttribute()
163 bool HasDeviceAllocatorAttribute(const AllocatorAttributes& attr) { in HasDeviceAllocatorAttribute()
170 AllocatorAttributes attr = DeviceAllocatorAttribute(); in TEST()
172 EXPECT_FALSE(HasDeviceAllocatorAttribute(AllocatorAttributes())); in TEST()
Ddevice_base.h182 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
190 Allocator* GetStepAllocator(AllocatorAttributes attr, ResourceMgr*) { in GetStepAllocator()
196 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
244 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Dop_kernel.h620 const AllocatorAttributes* output_attr_array = nullptr;
654 const gtl::InlinedVector<AllocatorAttributes, 4>* input_alloc_attrs =
853 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;
871 const TensorShape& shape, const AllocatorAttributes& allocator_attr,
878 AllocatorAttributes(), out_temp); in forward_input_or_allocate_temp()
958 AllocatorAttributes attr) TF_MUST_USE_RESULT;
961 AllocatorAttributes attr) TF_MUST_USE_RESULT;
968 Tensor* out_temp, AllocatorAttributes allocator_attr,
971 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()
977 return allocate_temp(type, shape, out_temp, AllocatorAttributes()); in allocate_temp()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.h33 Device* src, Device* dst, const AllocatorAttributes src_alloc_attr,
34 const AllocatorAttributes dst_alloc_attr, const Tensor* input,
45 const AllocatorAttributes src_alloc_attr,
46 const AllocatorAttributes dst_alloc_attr,
Dcollective_rma_local.h42 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
50 const AllocatorAttributes& from_alloc_attr,
78 Device* dst_dev, const AllocatorAttributes& src_attr,
79 const AllocatorAttributes& dst_attr,
Dthreadpool_device.h32 Allocator* GetAllocator(AllocatorAttributes attr) override;
33 Allocator* GetScopedAllocator(AllocatorAttributes attr,
39 const AllocatorAttributes alloc_attrs,
Dcollective_rma_local.cc29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
74 const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, in PostToPeer()
85 Device* dst_dev, const AllocatorAttributes& src_attr, in MemCpyAsync()
86 const AllocatorAttributes& dst_attr, const Tensor* src, Tensor* dst, in MemCpyAsync()
Dprocess_function_library_runtime.h61 const std::vector<AllocatorAttributes>& alloc_attrs,
74 const std::vector<AllocatorAttributes>& alloc_attrs,
147 using ArgAndRetAllocAttrs = std::pair<std::vector<AllocatorAttributes>,
148 std::vector<AllocatorAttributes>>;
170 std::vector<AllocatorAttributes> arg_alloc_attrs_;
173 std::vector<AllocatorAttributes> ret_alloc_attrs_;
Dthreadpool_device.cc76 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()
80 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
92 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Drendezvous_util.cc24 const std::vector<AllocatorAttributes>& alloc_attrs, in SendTensorsToRendezvous()
58 const std::vector<AllocatorAttributes>& alloc_attrs, in RecvOutputsFromRendezvousAsync()
73 std::tuple<string, Tensor*, Rendezvous::ParsedKey, AllocatorAttributes>> in RecvOutputsFromRendezvousAsync()
83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync()
Dpartitioning_utils.cc75 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata()
76 std::vector<AllocatorAttributes>* ret_alloc_attrs) { in UpdateArgAndRetvalMetadata()
102 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata()
113 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata()
Drenamed_device.h58 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
92 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Drendezvous_util.h35 const std::vector<AllocatorAttributes>& alloc_attrs,
44 const std::vector<AllocatorAttributes>& alloc_attrs,
Dpartitioning_utils.h62 std::vector<AllocatorAttributes>* arg_alloc_attrs,
63 std::vector<AllocatorAttributes>* ret_alloc_attrs);
Dbuf_rendezvous.h63 AllocatorAttributes prod_attr;
78 const Tensor* v, const AllocatorAttributes& attr,
Dsingle_threaded_cpu_device.cc66 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
76 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc38 Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()
46 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
48 AllocatorAttributes attr; in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc72 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
103 gtl::InlinedVector<AllocatorAttributes, 4> output_attrs; in EvaluateNode()
106 AllocatorAttributes attr; in EvaluateNode()
Devaluation_utils.h41 const AllocatorAttributes alloc_attrs,
44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h53 Allocator* GetAllocator(AllocatorAttributes attr) override;
60 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc30 typedef gtl::InlinedVector<AllocatorAttributes, 4> AllocatorAttributeVec;
124 AllocatorAttributes* attrs = kernel_state.output_alloc_attrs.data(); in Initialize()
131 AllocatorAttributes h; in Initialize()
341 std::vector<AllocatorAttributes>
349 std::vector<AllocatorAttributes>
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc47 AllocatorAttributes alloc_attrs;
194 AllocatorAttributes alloc_attr; in Compute()
236 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
252 AllocatorAttributes host_alloc_attrs; in ComputeAsync()
263 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
Dops_testutil.h53 std::vector<AllocatorAttributes>* attrs) { in SetOutputAttrs()
56 AllocatorAttributes attr; in SetOutputAttrs()
77 allocator_ = device_->GetAllocator(AllocatorAttributes());
174 std::vector<AllocatorAttributes> attrs; in RunOpKernel()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.h47 void InitAlloc(DeviceBase* d, const AllocatorAttributes& aa);
102 AllocatorAttributes alloc_attrs_;
Dcollective_rma_distributed.cc38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall()
78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
129 AllocatorAttributes cpu_attr; in RecvFromPeer()

123456