Searched refs:cpu_function_runtime (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | cpu_function_runtime_test.cc | 24 using cpu_function_runtime::BufferInfo;
32 EXPECT_EQ(cpu_function_runtime::kAlign, Allocator::kAllocatorAlignment); in TEST()
94 cpu_function_runtime::FreeContiguous(base); in TEST()
102 cpu_function_runtime::FreeContiguous(base); in TEST()
114 cpu_function_runtime::FreeContiguous(base); in TEST()
126 cpu_function_runtime::FreeContiguous(base); in TEST()
149 cpu_function_runtime::FreeContiguous(base); in TEST()
|
| D | xla_compiled_cpu_function.h | 69 const cpu_function_runtime::BufferInfo* buffer_infos_ = nullptr;
262 const cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos()
326 const cpu_function_runtime::BufferInfo* const buffer_infos_;
|
| D | xla_compiled_cpu_function.cc | 37 alloc_buffer_table_ = cpu_function_runtime::MallocContiguousBuffers( in XlaCompiledCpuFunction()
58 cpu_function_runtime::FreeContiguous(alloc_buffer_table_); in ~XlaCompiledCpuFunction()
|
| D | BUILD | 106 name = "cpu_function_runtime",
107 srcs = ["cpu_function_runtime.cc"],
108 hdrs = ["cpu_function_runtime.h"],
129 ":cpu_function_runtime",
139 ":cpu_function_runtime",
|
| D | xla_jit_compiled_cpu_function.h | 70 std::vector<cpu_function_runtime::BufferInfo> buffer_infos_;
|
| D | cpu_function_runtime.cc | 57 namespace cpu_function_runtime { namespace
|
| D | cpu_function_runtime.h | 24 namespace cpu_function_runtime {
|
| D | xla_jit_compiled_cpu_function.cc | 123 std::vector<cpu_function_runtime::BufferInfo> buffer_infos = in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | buffer_info_util.h | 27 std::vector<::tensorflow::cpu_function_runtime::BufferInfo>
37 absl::Span<const ::tensorflow::cpu_function_runtime::BufferInfo>
|
| D | cpu_compiler.h | 84 std::vector<::tensorflow::cpu_function_runtime::BufferInfo> buffer_infos,
94 const std::vector<::tensorflow::cpu_function_runtime::BufferInfo>&
106 const std::vector<::tensorflow::cpu_function_runtime::BufferInfo>
|
| D | buffer_info_util.cc | 21 using BufferInfo = ::tensorflow::cpu_function_runtime::BufferInfo;
|
| D | BUILD | 65 "//tensorflow/compiler/tf2xla:cpu_function_runtime",
96 "//tensorflow/compiler/tf2xla:cpu_function_runtime",
|
| D | cpu_compiler.cc | 115 using BufferInfo = ::tensorflow::cpu_function_runtime::BufferInfo;
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test_h.golden | 273 static const ::tensorflow::cpu_function_runtime::BufferInfo* BufferInfos() {
274 static const ::tensorflow::cpu_function_runtime::BufferInfo
276 ::tensorflow::cpu_function_runtime::BufferInfo({5ULL, ~0ULL}),
277 ::tensorflow::cpu_function_runtime::BufferInfo({34ULL, 0ULL}),
278 ::tensorflow::cpu_function_runtime::BufferInfo({9ULL, ~0ULL}),
279 ::tensorflow::cpu_function_runtime::BufferInfo({386ULL, 1ULL}),
280 ::tensorflow::cpu_function_runtime::BufferInfo({13ULL, ~0ULL}),
281 ::tensorflow::cpu_function_runtime::BufferInfo({481ULL, ~0ULL})
|
| D | codegen.cc | 41 using BufferInfo = cpu_function_runtime::BufferInfo;
401 const size_t arg_bytes_aligned = cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()
405 const size_t temp_bytes_aligned = cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()
|
| D | codegen_test.cc | 37 using ::tensorflow::cpu_function_runtime::BufferInfo;
|
| D | BUILD | 36 "//tensorflow/compiler/tf2xla:cpu_function_runtime",
|