Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc49 static_assert(std::is_trivially_destructible<AllocatorAttributes>::value, in ~GraphView()
98 + num_outputs * sizeof(AllocatorAttributes) // output_attr[...] in NodeItemBytes()
109 static_assert(kItemAlignment % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
113 static_assert(sizeof(NodeItem) % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
115 static_assert(sizeof(EdgeInfo) % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
187 AllocatorAttributes* output_attrs = item->output_attr_base(); in InitializeNode()
189 new (&output_attrs[i]) AllocatorAttributes(); in InitializeNode()
281 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
298 AllocatorAttributes* sa_attrs = sa_item->output_attr_base(); in SetScopedAllocatorAttrs()
307 AllocatorAttributes* use_attrs = item->output_attr_base(); in SetScopedAllocatorAttrs()
[all …]
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,
Dgraph_view.h136 const AllocatorAttributes* output_attrs() const { return output_attr_base(); } in output_attrs()
176 AllocatorAttributes* output_attr_base() const { in output_attr_base()
177 return reinterpret_cast<AllocatorAttributes*>( in output_attr_base()
185 sizeof(AllocatorAttributes) * num_outputs); in forward_from_base()
191 sizeof(AllocatorAttributes) * num_outputs + sizeof(int) * num_outputs); in input_type_base()
197 sizeof(AllocatorAttributes) * num_outputs + sizeof(int) * num_outputs + in output_type_base()
Dcollective_rma_local.h43 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
52 const AllocatorAttributes& from_alloc_attr,
68 Device* dst_dev, const AllocatorAttributes& src_attr,
69 const AllocatorAttributes& dst_attr,
Dthreadpool_device.h32 Allocator* GetAllocator(AllocatorAttributes attr) override;
33 Allocator* GetScopedAllocator(AllocatorAttributes attr,
39 const AllocatorAttributes alloc_attrs,
Dthreadpool_device.cc73 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()
77 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
89 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Dcollective_rma_local.cc29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
100 const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, in PostToPeer()
120 Device* dst_dev, const AllocatorAttributes& src_attr, in MemCpyAsync()
121 const AllocatorAttributes& dst_attr, const Tensor* src, Tensor* dst, in MemCpyAsync()
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()
Dcollective_rma_local_test.cc70 AllocatorAttributes attr; in TEST_F()
113 AllocatorAttributes attr; in TEST_F()
170 AllocatorAttributes attr; in TEST_F()
192 AllocatorAttributes attr; in TEST_F()
214 AllocatorAttributes attr; in TEST_F()
235 AllocatorAttributes attr; in TEST_F()
Drendezvous_util.h35 const std::vector<AllocatorAttributes>& alloc_attrs,
44 const std::vector<AllocatorAttributes>& alloc_attrs,
Dpartitioning_utils.h66 std::vector<AllocatorAttributes>* arg_alloc_attrs,
67 std::vector<AllocatorAttributes>* ret_alloc_attrs);
Dpartitioning_utils.cc79 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata()
80 std::vector<AllocatorAttributes>* ret_alloc_attrs) { in UpdateArgAndRetvalMetadata()
128 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata()
145 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata()
Drenamed_device.h57 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
61 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
107 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc53 AllocatorAttributes aa; in TEST()
66 AllocatorAttributes a, b; in TEST()
90 AllocatorAttributes a, b; in TEST()
127 AllocatorAttributes a, b; in TEST()
206 AllocatorAttributes DeviceAllocatorAttribute() { in DeviceAllocatorAttribute()
207 AllocatorAttributes attr; in DeviceAllocatorAttribute()
212 bool HasDeviceAllocatorAttribute(const AllocatorAttributes& attr) { in HasDeviceAllocatorAttribute()
219 AllocatorAttributes attr = DeviceAllocatorAttribute(); in TEST()
221 EXPECT_FALSE(HasDeviceAllocatorAttribute(AllocatorAttributes())); in TEST()
Ddevice_base.h181 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
189 Allocator* GetStepAllocator(AllocatorAttributes attr, ResourceMgr*) { in GetStepAllocator()
195 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
239 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Dop_kernel.h289 Tensor* out_temp, AllocatorAttributes allocator_attr);
637 const AllocatorAttributes* output_attr_array = nullptr;
674 const gtl::InlinedVector<AllocatorAttributes, 4>* input_alloc_attrs =
894 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;
916 const TensorShape& shape, const AllocatorAttributes& allocator_attr,
923 AllocatorAttributes(), out_temp); in forward_input_or_allocate_temp()
1002 AllocatorAttributes attr) TF_MUST_USE_RESULT;
1005 AllocatorAttributes attr) TF_MUST_USE_RESULT;
1012 Tensor* out_temp, AllocatorAttributes allocator_attr,
1015 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_framework_c_interface.cc28 using tensorflow::AllocatorAttributes;
33 AllocatorAttributes attrs; in GetAllocator()
56 AllocatorAttributes output_attr; in _mlir_ciface_tf_alloc()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h144 Allocator* GetAllocator(AllocatorAttributes attr) override
156 const AllocatorAttributes alloc_attrs,
163 const AllocatorAttributes alloc_attrs,
193 Allocator* GetAllocatorLocked(AllocatorAttributes attr)
207 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.cc57 std::vector<AllocatorAttributes>* attrs) { in SetOutputAttrs()
60 AllocatorAttributes attr; in SetOutputAttrs()
81 allocator_ = device_->GetAllocator(AllocatorAttributes()); in OpsTestBase()
121 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
127 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
173 std::vector<AllocatorAttributes> attrs; in RunOpKernel()
Dstack.cc47 AllocatorAttributes alloc_attrs;
195 AllocatorAttributes alloc_attr; in Compute()
237 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
253 AllocatorAttributes host_alloc_attrs; in ComputeAsync()
264 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc51 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
82 gtl::InlinedVector<AllocatorAttributes, 4> output_attrs; in EvaluateNode()
85 AllocatorAttributes attr; in EvaluateNode()
Devaluation_utils.h41 const AllocatorAttributes alloc_attrs,
44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.h30 Allocator* GetAllocator(AllocatorAttributes attr) override;
32 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h53 Allocator* GetAllocator(AllocatorAttributes attr) override;
60 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.h47 void InitAlloc(DeviceBase* d, const AllocatorAttributes& aa);
102 AllocatorAttributes alloc_attrs_;

1234567