Home
last modified time | relevance | path

Searched defs:buffer_infos (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc25 std::vector<BufferInfo> buffer_infos; in CreateBufferInfosFromBufferAssignment() local
43 absl::Span<const BufferInfo> buffer_infos) { in CreateArgIndexTableFromBufferInfos()
Dcpu_compiler.h95 buffer_infos() const { in buffer_infos() function
Dcpu_compiler.cc133 ObjectFileData object_file_data, std::vector<BufferInfo> buffer_infos, in CpuAotCompilationResult()
927 std::vector<BufferInfo> buffer_infos = in CompileAheadOfTime() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime_test.cc36 std::vector<BufferInfo> buffer_infos; in SizesToBufferInfos() local
53 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in AlignedBufferBytesFromSizes() local
60 std::vector<BufferInfo> buffer_infos = SizesToBufferInfos(sizes, n); in MallocContiguousBuffersFromSizes() local
Dcpu_function_runtime.cc58 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n, in AlignedBufferBytes()
73 void* MallocContiguousBuffers(const BufferInfo* buffer_infos, size_t n, in MallocContiguousBuffers()
Dxla_jit_compiled_cpu_function.cc123 std::vector<cpu_function_runtime::BufferInfo> buffer_infos = in Compile() local
Dxla_compiled_cpu_function.h262 const cpu_function_runtime::BufferInfo* buffer_infos) { in set_static_data_buffer_infos()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen.cc93 size_t TotalBufferBytes(const std::vector<BufferInfo>& buffer_infos) { in TotalBufferBytes()
103 const std::vector<BufferInfo>& buffer_infos) { in ExtractEntryParamBufferInfos()
115 const std::vector<BufferInfo>& buffer_infos) { in ExtractTempBufferInfos()
355 const std::vector<BufferInfo>& buffer_infos) { in BufferInfosToCppExpression()
379 const std::vector<BufferInfo>& buffer_infos = in GenerateHeader() local