Home
last modified time | relevance | path

Searched refs:buffer_infos (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc26 std::vector<BufferInfo> buffer_infos; in CreateBufferInfosFromBufferAssignment() local
29 buffer_infos.push_back(BufferInfo::MakeOnStackBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()
31 buffer_infos.push_back(BufferInfo::MakeConstant(allocation.size())); in CreateBufferInfosFromBufferAssignment()
33 buffer_infos.push_back(BufferInfo::MakeEntryParameter( in CreateBufferInfosFromBufferAssignment()
37 buffer_infos.push_back(BufferInfo::MakeTempBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()
40 return buffer_infos; in CreateBufferInfosFromBufferAssignment()
44 absl::Span<const BufferInfo> buffer_infos) { in CreateArgIndexTableFromBufferInfos() argument
46 for (int64 i = 0; i < buffer_infos.size(); i++) { in CreateArgIndexTableFromBufferInfos()
47 if (buffer_infos[i].is_entry_parameter()) { in CreateArgIndexTableFromBufferInfos()
48 if (buffer_infos[i].entry_parameter_number() >= result.size()) { in CreateArgIndexTableFromBufferInfos()
[all …]
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() function
Dbuffer_info_util.h37 absl::Span<const cpu_function_runtime::BufferInfo> buffer_infos);
Dcpu_compiler.cc171 ObjectFileData object_file_data, std::vector<BufferInfo> buffer_infos, in CpuAotCompilationResult() argument
175 buffer_infos_(std::move(buffer_infos)), in CpuAotCompilationResult()
1013 std::vector<BufferInfo> buffer_infos = in CompileAheadOfTime() local
1020 std::move(object_file_data), std::move(buffer_infos), in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/
Dcpu_function_runtime.cc58 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()
Dcpu_function_runtime.h159 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n,
174 void* MallocContiguousBuffers(const BufferInfo* buffer_infos, size_t n,
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen.cc95 size_t TotalBufferBytes(const std::vector<BufferInfo>& buffer_infos) { in TotalBufferBytes() argument
96 return std::accumulate(buffer_infos.begin(), buffer_infos.end(), size_t{0}, in TotalBufferBytes()
105 const std::vector<BufferInfo>& buffer_infos) { in ExtractEntryParamBufferInfos() argument
107 std::copy_if(buffer_infos.begin(), buffer_infos.end(), in ExtractEntryParamBufferInfos()
117 const std::vector<BufferInfo>& buffer_infos) { in ExtractTempBufferInfos() argument
119 std::copy_if(buffer_infos.begin(), buffer_infos.end(), in ExtractTempBufferInfos()
375 const std::vector<BufferInfo>& buffer_infos) { in BufferInfosToCppExpression() argument
377 std::transform(buffer_infos.begin(), buffer_infos.end(), in BufferInfosToCppExpression()
399 const std::vector<BufferInfo>& buffer_infos = in GenerateHeader() local
400 compile_result.aot->buffer_infos(); in GenerateHeader()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime_test.cc37 std::vector<BufferInfo> buffer_infos; in SizesToBufferInfos() local
38 std::transform(sizes, sizes + n, std::back_inserter(buffer_infos), in SizesToBufferInfos()
48 return buffer_infos; in SizesToBufferInfos()
54 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in AlignedBufferBytesFromSizes() local
55 return AlignedBufferBytes(buffer_infos.data(), n, in AlignedBufferBytesFromSizes()
61 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in MallocContiguousBuffersFromSizes() local
62 return MallocContiguousBuffers(buffer_infos.data(), n, in MallocContiguousBuffersFromSizes()
Dxla_jit_compiled_cpu_function.cc132 std::vector<xla::cpu_function_runtime::BufferInfo> buffer_infos = in Compile() local
135 xla::cpu::CreateArgIndexTableFromBufferInfos(buffer_infos); in Compile()
143 jit->buffer_infos_ = std::move(buffer_infos); in Compile()
Dxla_compiled_cpu_function.h282 const xla::cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos() argument
283 static_data->buffer_infos_ = buffer_infos; in set_static_data_buffer_infos()
/external/deqp-deps/amber/src/vulkan/
Dbuffer_descriptor.cc89 std::vector<VkDescriptorBufferInfo> buffer_infos; in UpdateDescriptorSetIfNeeded() local
97 buffer_infos.push_back(buffer_info); in UpdateDescriptorSetIfNeeded()
109 write.descriptorCount = static_cast<uint32_t>(buffer_infos.size()); in UpdateDescriptorSetIfNeeded()
111 write.pBufferInfo = buffer_infos.data(); in UpdateDescriptorSetIfNeeded()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_aot_test_helper.cc102 CHECK_EQ(result->buffer_infos().size(), 3); in main()
103 CHECK(result->buffer_infos()[0].is_entry_parameter()); // param buffer in main()
104 CHECK_EQ(result->buffer_infos()[1].size(), sizeof(float)); // result buffer in main()
105 CHECK(result->buffer_infos()[2].is_constant()); // const buffer in main()
/external/mesa3d/src/gallium/drivers/zink/
Dzink_draw.c291 VkDescriptorBufferInfo buffer_infos[PIPE_SHADER_TYPES * PIPE_MAX_CONSTANT_BUFFERS]; in zink_draw_vbo() local
322 buffer_infos[num_buffer_info].buffer = res ? res->buffer : in zink_draw_vbo()
326 buffer_infos[num_buffer_info].offset = res ? ctx->ubos[i][index].buffer_offset : 0; in zink_draw_vbo()
327 … buffer_infos[num_buffer_info].range = res ? ctx->ubos[i][index].buffer_size : VK_WHOLE_SIZE; in zink_draw_vbo()
328 wds[num_wds].pBufferInfo = buffer_infos + num_buffer_info; in zink_draw_vbo()
/external/vulkan-validation-layers/tests/
Dlayer_validation_tests.cpp1360 buffer_infos.emplace_back(buffer_info); in WriteDescriptorBufferInfo()
1361 size_t index = buffer_infos.size() - 1; in WriteDescriptorBufferInfo()
1370 descriptor_write.pBufferInfo = &buffer_infos[index]; in WriteDescriptorBufferInfo()
Dlayer_validation_tests.h399 std::vector<VkDescriptorBufferInfo> buffer_infos; member