Home
last modified time | relevance | path

Searched refs:cpu_function_runtime (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dcpu_function_runtime_test.cc23 using ::xla::cpu_function_runtime::BufferInfo;
31 EXPECT_EQ(xla::cpu_function_runtime::Align(), Allocator::kAllocatorAlignment); in TEST()
32 EXPECT_LE(xla::cpu_function_runtime::MinAlign(), in TEST()
95 xla::cpu_function_runtime::FreeContiguous(base); in TEST()
103 xla::cpu_function_runtime::FreeContiguous(base); in TEST()
115 xla::cpu_function_runtime::FreeContiguous(base); in TEST()
127 xla::cpu_function_runtime::FreeContiguous(base); in TEST()
150 xla::cpu_function_runtime::FreeContiguous(base); in TEST()
H A Dxla_compiled_cpu_function.h70 const xla::cpu_function_runtime::BufferInfo* buffer_infos_ = nullptr;
183 assert((arg_size(index) < xla::cpu_function_runtime::MinAlign() || in set_arg_data()
184 (uintptr_t)data % xla::cpu_function_runtime::MinAlign() == 0) && in set_arg_data()
283 const xla::cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos()
357 const xla::cpu_function_runtime::BufferInfo* const buffer_infos_;
H A Dxla_compiled_cpu_function.cc41 alloc_buffer_table_ = xla::cpu_function_runtime::MallocContiguousBuffers( in XlaCompiledCpuFunction()
63 xla::cpu_function_runtime::FreeContiguous(alloc_buffer_table_); in ~XlaCompiledCpuFunction()
H A Dxla_jit_compiled_cpu_function.h71 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos_;
H A Dxla_jit_compiled_cpu_function.cc132 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos = in Compile()
H A DBUILD343 "//tensorflow/compiler/xla:cpu_function_runtime",
353 "//tensorflow/compiler/xla:cpu_function_runtime",
369 "//tensorflow/compiler/xla:cpu_function_runtime",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/
H A Dcodegen_test_h.golden354 static const ::xla::cpu_function_runtime::BufferInfo* BufferInfos() {
355 static const ::xla::cpu_function_runtime::BufferInfo
357 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
358 ::xla::cpu_function_runtime::BufferInfo({34ULL, 0ULL}),
359 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
360 ::xla::cpu_function_runtime::BufferInfo({386ULL, 1ULL}),
361 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
362 ::xla::cpu_function_runtime::BufferInfo({386ULL, 2ULL}),
363 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
364 ::xla::cpu_function_runtime::BufferInfo({386ULL, 3ULL}),
[all …]
H A Dcodegen.cc43 using BufferInfo = xla::cpu_function_runtime::BufferInfo;
423 xla::cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()
428 xla::cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()
H A DBUILD58 "//tensorflow/compiler/xla:cpu_function_runtime",
96 "//tensorflow/compiler/xla:cpu_function_runtime",
H A Dcodegen_test.cc40 using ::xla::cpu_function_runtime::BufferInfo;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.h90 std::vector<cpu_function_runtime::BufferInfo> buffer_infos,
100 const std::vector<cpu_function_runtime::BufferInfo>& buffer_infos() const { in buffer_infos()
111 const std::vector<cpu_function_runtime::BufferInfo> buffer_infos_;
H A Dbuffer_info_util.h27 std::vector<cpu_function_runtime::BufferInfo>
37 absl::Span<const cpu_function_runtime::BufferInfo> buffer_infos);
H A Dtarget_machine_features.cc51 cpu_function_runtime::MinAlign()); in minimum_alignment_for_allocation()
H A Dbuffer_info_util.cc23 using BufferInfo = cpu_function_runtime::BufferInfo;
H A DBUILD168 "//tensorflow/compiler/xla:cpu_function_runtime",
263 "//tensorflow/compiler/xla:cpu_function_runtime",
561 "//tensorflow/compiler/xla:cpu_function_runtime",
H A Dcpu_compiler.cc270 using BufferInfo = cpu_function_runtime::BufferInfo;
713 return cpu_function_runtime::MinAlign(); in memory_alignment()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dcpu_function_runtime.h26 namespace cpu_function_runtime {
150 #define XLA_ALIGN alignas(xla::cpu_function_runtime::Align())
H A DBUILD52 "cpu_function_runtime.cc",
61 "cpu_function_runtime.h",
985 name = "cpu_function_runtime",
986 srcs = ["cpu_function_runtime.cc"],
987 hdrs = ["cpu_function_runtime.h"],
H A Dcpu_function_runtime.cc57 namespace cpu_function_runtime { namespace
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_tfrt_cpu_device_buffer.h58 size, cpu_function_runtime::MinAlign())); in AllocateShared()
H A DBUILD305 "//tensorflow/compiler/xla:cpu_function_runtime",
534 "//tensorflow/compiler/xla:cpu_function_runtime",
H A Dpjrt_stream_executor_client.cc736 (cpu_function_runtime::MinAlign() - 1)) == 0); in BufferFromHostBuffer()
753 cpu_function_runtime::MinAlign(), size); in BufferFromHostBuffer()
H A Dtfrt_cpu_pjrt_client.cc510 (cpu_function_runtime::MinAlign() - 1)) == 0); in BufferFromHostBuffer()