Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc51 static_assert(std::is_trivially_destructible<AllocatorAttributes>::value, in ~GraphView()
100 + num_outputs * sizeof(AllocatorAttributes) // output_attr[...] in NodeItemBytes()
111 static_assert(kItemAlignment % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
115 static_assert(sizeof(NodeItem) % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
117 static_assert(sizeof(EdgeInfo) % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
189 AllocatorAttributes* output_attrs = item->output_attr_base(); in InitializeNode()
191 new (&output_attrs[i]) AllocatorAttributes(); in InitializeNode()
298 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr()
315 AllocatorAttributes* sa_attrs = sa_item->output_attr_base(); in SetScopedAllocatorAttrs()
324 AllocatorAttributes* use_attrs = item->output_attr_base(); in SetScopedAllocatorAttrs()
[all …]
Dgraph_view.h138 const AllocatorAttributes* output_attrs() const { return output_attr_base(); } in output_attrs()
178 AllocatorAttributes* output_attr_base() const { in output_attr_base()
179 return reinterpret_cast<AllocatorAttributes*>( in output_attr_base()
187 sizeof(AllocatorAttributes) * num_outputs); in forward_from_base()
193 sizeof(AllocatorAttributes) * num_outputs + sizeof(int) * num_outputs); in input_type_base()
199 sizeof(AllocatorAttributes) * num_outputs + sizeof(int) * num_outputs + in output_type_base()
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.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.h33 Allocator* GetAllocator(AllocatorAttributes attr) override;
34 Allocator* GetScopedAllocator(AllocatorAttributes attr,
40 const AllocatorAttributes alloc_attrs,
Dpartitioning_utils_test.cc270 const std::vector<AllocatorAttributes>& actual) { in CheckAlloc()
290 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F()
291 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F()
316 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F()
317 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F()
334 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F()
335 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F()
363 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F()
364 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F()
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.cc71 AllocatorAttributes attr; in TEST_F()
114 AllocatorAttributes attr; in TEST_F()
171 AllocatorAttributes attr; in TEST_F()
193 AllocatorAttributes attr; in TEST_F()
215 AllocatorAttributes attr; in TEST_F()
236 AllocatorAttributes attr; in TEST_F()
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()
Dcollective_test_util.cc61 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
74 const AllocatorAttributes& from_alloc_attr, in PostToPeer()
266 Tensor copied(device->GetAllocator(AllocatorAttributes()), tensor.dtype(), in CopyTensorToDevice()
306 output_buffer = Tensor(device->GetAllocator(AllocatorAttributes()), in RunCollective()
325 gtl::InlinedVector<AllocatorAttributes, 4> input_aa({AllocatorAttributes()}); in RunCollective()
339 AllocatorAttributes generic_alloc_attr; in RunCollective()
Drendezvous_util.h35 const std::vector<AllocatorAttributes>& alloc_attrs,
44 const std::vector<AllocatorAttributes>& alloc_attrs,
Drenamed_device.h58 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator()
62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
108 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc59 AllocatorAttributes aa; in TEST()
72 AllocatorAttributes a, b; in TEST()
96 AllocatorAttributes a, b; in TEST()
133 AllocatorAttributes a, b; in TEST()
310 AllocatorAttributes DeviceAllocatorAttribute() { in DeviceAllocatorAttribute()
311 AllocatorAttributes attr; in DeviceAllocatorAttribute()
316 bool HasDeviceAllocatorAttribute(const AllocatorAttributes& attr) { in HasDeviceAllocatorAttribute()
323 AllocatorAttributes attr = DeviceAllocatorAttribute(); in TEST()
325 EXPECT_FALSE(HasDeviceAllocatorAttribute(AllocatorAttributes())); in TEST()
Ddevice_base.h185 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
193 Allocator* GetStepAllocator(AllocatorAttributes attr, ResourceMgr*) { in GetStepAllocator()
199 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
251 const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Dop_kernel.h268 Tensor* out_temp, AllocatorAttributes allocator_attr);
610 const AllocatorAttributes* output_attr_array = nullptr;
647 absl::Span<const AllocatorAttributes> input_alloc_attrs;
875 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;
897 const TensorShape& shape, const AllocatorAttributes& allocator_attr,
904 AllocatorAttributes(), out_temp); in forward_input_or_allocate_temp()
981 AllocatorAttributes attr) TF_MUST_USE_RESULT;
984 AllocatorAttributes attr) TF_MUST_USE_RESULT;
991 Tensor* out_temp, AllocatorAttributes allocator_attr,
994 Tensor* out_temp, AllocatorAttributes allocator_attr);
[all …]
/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()
115 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
121 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
173 std::vector<AllocatorAttributes> attrs; in RunOpKernel()
/external/tensorflow/tensorflow/core/tfrt/fallback/
Dop_kernel_runner.h76 const gtl::InlinedVector<AllocatorAttributes, 4>& input_alloc_attrs() const { in input_alloc_attrs()
79 const gtl::InlinedVector<AllocatorAttributes, 1>& output_alloc_attrs() const { in output_alloc_attrs()
94 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_;
95 gtl::InlinedVector<AllocatorAttributes, 1> output_alloc_attrs_;
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dop_kernel_runner.h95 const gtl::InlinedVector<AllocatorAttributes, 4>& input_alloc_attrs() const { in input_alloc_attrs()
98 const gtl::InlinedVector<AllocatorAttributes, 1>& output_alloc_attrs() const { in output_alloc_attrs()
113 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_;
114 gtl::InlinedVector<AllocatorAttributes, 1> output_alloc_attrs_;
/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/common_runtime/pluggable_device/
Dpluggable_device.h66 Allocator* GetAllocator(AllocatorAttributes attr) override;
69 const AllocatorAttributes alloc_attrs,
112 const AllocatorAttributes& alloc_attrs, const Tensor& from, Tensor* to,
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.h30 Allocator* GetAllocator(AllocatorAttributes attr) override;
32 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h154 Allocator* GetAllocator(AllocatorAttributes attr) override
166 const AllocatorAttributes alloc_attrs,
171 const AllocatorAttributes alloc_attrs,
208 Allocator* GetAllocatorLocked(AllocatorAttributes attr)
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h53 Allocator* GetAllocator(AllocatorAttributes attr) override;
60 const AllocatorAttributes alloc_attrs,
/external/tensorflow/tensorflow/core/transforms/utils/
Deval_utils.h46 const tensorflow::AllocatorAttributes alloc_attrs,
50 tensorflow::AllocatorAttributes attr) override;

12345678