Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime_test.cc23 using ::xla::cpu_function_runtime::BufferInfo;
31 EXPECT_EQ(xla::cpu_function_runtime::kAlign, Allocator::kAllocatorAlignment); in TEST()
32 EXPECT_LE(xla::cpu_function_runtime::kMinAlign, 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()
Dxla_compiled_cpu_function.h69 const xla::cpu_function_runtime::BufferInfo* buffer_infos_ = nullptr;
182 assert((arg_size(index) < xla::cpu_function_runtime::kMinAlign || in set_arg_data()
183 (uintptr_t)data % xla::cpu_function_runtime::kMinAlign == 0) && in set_arg_data()
282 const xla::cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos()
356 const xla::cpu_function_runtime::BufferInfo* const buffer_infos_;
Dxla_compiled_cpu_function.cc40 alloc_buffer_table_ = xla::cpu_function_runtime::MallocContiguousBuffers( in XlaCompiledCpuFunction()
61 xla::cpu_function_runtime::FreeContiguous(alloc_buffer_table_); in ~XlaCompiledCpuFunction()
Dxla_jit_compiled_cpu_function.h71 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos_;
Dxla_jit_compiled_cpu_function.cc132 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos = in Compile()
DBUILD269 "//tensorflow/compiler/xla:cpu_function_runtime",
279 "//tensorflow/compiler/xla:cpu_function_runtime",
295 "//tensorflow/compiler/xla:cpu_function_runtime",
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test_h.golden353 static const ::xla::cpu_function_runtime::BufferInfo* BufferInfos() {
354 static const ::xla::cpu_function_runtime::BufferInfo
356 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
357 ::xla::cpu_function_runtime::BufferInfo({34ULL, 0ULL}),
358 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
359 ::xla::cpu_function_runtime::BufferInfo({386ULL, 1ULL}),
360 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
361 ::xla::cpu_function_runtime::BufferInfo({386ULL, 2ULL}),
362 ::xla::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
363 ::xla::cpu_function_runtime::BufferInfo({386ULL, 3ULL}),
[all …]
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()
DBUILD57 "//tensorflow/compiler/xla:cpu_function_runtime",
95 "//tensorflow/compiler/xla:cpu_function_runtime",
Dcodegen_test.cc40 using ::xla::cpu_function_runtime::BufferInfo;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.h27 std::vector<cpu_function_runtime::BufferInfo>
37 absl::Span<const cpu_function_runtime::BufferInfo> buffer_infos);
Dcpu_compiler.h85 std::vector<cpu_function_runtime::BufferInfo> buffer_infos,
95 const std::vector<cpu_function_runtime::BufferInfo>& buffer_infos() const { in buffer_infos()
106 const std::vector<cpu_function_runtime::BufferInfo> buffer_infos_;
Dtarget_machine_features.cc49 cpu_function_runtime::kMinAlign); in minimum_alignment_for_allocation()
Dbuffer_info_util.cc22 using BufferInfo = cpu_function_runtime::BufferInfo;
Dcpu_compiler.cc153 using BufferInfo = cpu_function_runtime::BufferInfo;
468 return cpu_function_runtime::kMinAlign; in memory_alignment()
DBUILD123 "//tensorflow/compiler/xla:cpu_function_runtime",
173 "//tensorflow/compiler/xla:cpu_function_runtime",
431 "//tensorflow/compiler/xla:cpu_function_runtime",
/external/tensorflow/tensorflow/compiler/xla/
Dcpu_function_runtime.h24 namespace cpu_function_runtime {
153 #define XLA_ALIGN alignas(xla::cpu_function_runtime::kAlign)
DBUILD43 "cpu_function_runtime.cc",
52 "cpu_function_runtime.h",
937 name = "cpu_function_runtime",
938 srcs = ["cpu_function_runtime.cc"],
939 hdrs = ["cpu_function_runtime.h"],
Dcpu_function_runtime.cc57 namespace cpu_function_runtime { namespace
/external/tensorflow/tensorflow/compiler/xla/pjrt/
DBUILD184 "//tensorflow/compiler/xla:cpu_function_runtime",
Dpjrt_stream_executor_client.cc681 (cpu_function_runtime::kMinAlign - 1)) == 0); in BufferFromHostBuffer()
697 cpu_function_runtime::kMinAlign, size); in BufferFromHostBuffer()