| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | buffer_info_util.cc | 25 std::vector<BufferInfo> buffer_infos; in CreateBufferInfosFromBufferAssignment() local
28 buffer_infos.push_back(BufferInfo::MakeOnStackBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()
30 buffer_infos.push_back(BufferInfo::MakeConstant(allocation.size())); in CreateBufferInfosFromBufferAssignment()
32 buffer_infos.push_back(BufferInfo::MakeEntryParameter( in CreateBufferInfosFromBufferAssignment()
36 buffer_infos.push_back(BufferInfo::MakeTempBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()
39 return buffer_infos; in CreateBufferInfosFromBufferAssignment()
43 absl::Span<const BufferInfo> buffer_infos) { in CreateArgIndexTableFromBufferInfos() argument
45 for (int64 i = 0; i < buffer_infos.size(); i++) { in CreateArgIndexTableFromBufferInfos()
46 if (buffer_infos[i].is_entry_parameter()) { in CreateArgIndexTableFromBufferInfos()
47 if (buffer_infos[i].entry_parameter_number() >= result.size()) { in CreateArgIndexTableFromBufferInfos()
[all …]
|
| D | cpu_compiler.h | 84 std::vector<::tensorflow::cpu_function_runtime::BufferInfo> buffer_infos,
95 buffer_infos() const { in buffer_infos() function
|
| D | buffer_info_util.h | 38 buffer_infos);
|
| D | cpu_compiler.cc | 133 ObjectFileData object_file_data, std::vector<BufferInfo> buffer_infos, in CpuAotCompilationResult() argument
137 buffer_infos_(std::move(buffer_infos)), in CpuAotCompilationResult()
927 std::vector<BufferInfo> buffer_infos = in CompileAheadOfTime() local
934 std::move(object_file_data), std::move(buffer_infos), in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | cpu_function_runtime.cc | 58 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n, in AlignedBufferBytes() argument
63 buffer_infos[i].is_temp_buffer() || in AlignedBufferBytes()
64 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in AlignedBufferBytes()
67 total += align_to(buffer_infos[i].size(), kAlign); in AlignedBufferBytes()
73 void* MallocContiguousBuffers(const BufferInfo* buffer_infos, size_t n, in MallocContiguousBuffers() argument
77 AlignedBufferBytes(buffer_infos, n, allocate_entry_params); in MallocContiguousBuffers()
90 buffer_infos[i].is_temp_buffer() || in MallocContiguousBuffers()
91 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in MallocContiguousBuffers()
94 pos += align_to(buffer_infos[i].size(), kAlign); in MallocContiguousBuffers()
|
| D | cpu_function_runtime_test.cc | 36 std::vector<BufferInfo> buffer_infos; in SizesToBufferInfos() local
37 std::transform(sizes, sizes + n, std::back_inserter(buffer_infos), in SizesToBufferInfos()
47 return buffer_infos; in SizesToBufferInfos()
53 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in AlignedBufferBytesFromSizes() local
54 return AlignedBufferBytes(buffer_infos.data(), n, in AlignedBufferBytesFromSizes()
60 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in MallocContiguousBuffersFromSizes() local
61 return MallocContiguousBuffers(buffer_infos.data(), n, in MallocContiguousBuffersFromSizes()
|
| D | xla_jit_compiled_cpu_function.cc | 123 std::vector<cpu_function_runtime::BufferInfo> buffer_infos = in Compile() local
126 xla::cpu::CreateArgIndexTableFromBufferInfos(buffer_infos); in Compile()
134 jit->buffer_infos_ = std::move(buffer_infos); in Compile()
|
| D | cpu_function_runtime.h | 140 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n,
155 void* MallocContiguousBuffers(const BufferInfo* buffer_infos, size_t n,
|
| D | xla_compiled_cpu_function.h | 262 const cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos() argument
263 static_data->buffer_infos_ = buffer_infos; in set_static_data_buffer_infos()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 93 size_t TotalBufferBytes(const std::vector<BufferInfo>& buffer_infos) { in TotalBufferBytes() argument
94 return std::accumulate(buffer_infos.begin(), buffer_infos.end(), size_t{0}, in TotalBufferBytes()
103 const std::vector<BufferInfo>& buffer_infos) { in ExtractEntryParamBufferInfos() argument
105 std::copy_if(buffer_infos.begin(), buffer_infos.end(), in ExtractEntryParamBufferInfos()
115 const std::vector<BufferInfo>& buffer_infos) { in ExtractTempBufferInfos() argument
117 std::copy_if(buffer_infos.begin(), buffer_infos.end(), in ExtractTempBufferInfos()
355 const std::vector<BufferInfo>& buffer_infos) { in BufferInfosToCppExpression() argument
357 std::transform(buffer_infos.begin(), buffer_infos.end(), in BufferInfosToCppExpression()
379 const std::vector<BufferInfo>& buffer_infos = in GenerateHeader() local
380 compile_result.aot->buffer_infos(); in GenerateHeader()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_aot_test_helper.cc | 95 CHECK_EQ(result->buffer_infos().size(), 3); in main()
96 CHECK(result->buffer_infos()[0].is_entry_parameter()); // param buffer in main()
97 CHECK_EQ(result->buffer_infos()[1].size(), sizeof(float)); // result buffer in main()
98 CHECK(result->buffer_infos()[2].is_constant()); // const buffer in main()
|