/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | cpu_function_runtime_test.cc | 23 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 D | xla_compiled_cpu_function.h | 70 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 D | xla_compiled_cpu_function.cc | 41 alloc_buffer_table_ = xla::cpu_function_runtime::MallocContiguousBuffers( in XlaCompiledCpuFunction() 63 xla::cpu_function_runtime::FreeContiguous(alloc_buffer_table_); in ~XlaCompiledCpuFunction()
|
H A D | xla_jit_compiled_cpu_function.h | 71 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos_;
|
H A D | xla_jit_compiled_cpu_function.cc | 132 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos = in Compile()
|
H A D | BUILD | 343 "//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 D | codegen_test_h.golden | 354 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 D | codegen.cc | 43 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 D | BUILD | 58 "//tensorflow/compiler/xla:cpu_function_runtime", 96 "//tensorflow/compiler/xla:cpu_function_runtime",
|
H A D | codegen_test.cc | 40 using ::xla::cpu_function_runtime::BufferInfo;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_compiler.h | 90 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 D | buffer_info_util.h | 27 std::vector<cpu_function_runtime::BufferInfo> 37 absl::Span<const cpu_function_runtime::BufferInfo> buffer_infos);
|
H A D | target_machine_features.cc | 51 cpu_function_runtime::MinAlign()); in minimum_alignment_for_allocation()
|
H A D | buffer_info_util.cc | 23 using BufferInfo = cpu_function_runtime::BufferInfo;
|
H A D | BUILD | 168 "//tensorflow/compiler/xla:cpu_function_runtime", 263 "//tensorflow/compiler/xla:cpu_function_runtime", 561 "//tensorflow/compiler/xla:cpu_function_runtime",
|
H A D | cpu_compiler.cc | 270 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 D | cpu_function_runtime.h | 26 namespace cpu_function_runtime { 150 #define XLA_ALIGN alignas(xla::cpu_function_runtime::Align())
|
H A D | BUILD | 52 "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 D | cpu_function_runtime.cc | 57 namespace cpu_function_runtime { namespace
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
H A D | tracked_tfrt_cpu_device_buffer.h | 58 size, cpu_function_runtime::MinAlign())); in AllocateShared()
|
H A D | BUILD | 305 "//tensorflow/compiler/xla:cpu_function_runtime", 534 "//tensorflow/compiler/xla:cpu_function_runtime",
|
H A D | pjrt_stream_executor_client.cc | 736 (cpu_function_runtime::MinAlign() - 1)) == 0); in BufferFromHostBuffer() 753 cpu_function_runtime::MinAlign(), size); in BufferFromHostBuffer()
|
H A D | tfrt_cpu_pjrt_client.cc | 510 (cpu_function_runtime::MinAlign() - 1)) == 0); in BufferFromHostBuffer()
|