MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ExtractTempBufferInfos

Function ExtractTempBufferInfos

tensorflow/compiler/aot/codegen.cc:115–123  ·  view source on GitHub ↗

Returns a vector of BufferInfo instances in `buffer_infos` that are temp buffers.

Source from the content-addressed store, hash-verified

113// Returns a vector of BufferInfo instances in `buffer_infos` that are temp
114// buffers.
115std::vector<BufferInfo> ExtractTempBufferInfos(
116 const std::vector<BufferInfo>& buffer_infos) {
117 std::vector<BufferInfo> result;
118 std::copy_if(buffer_infos.begin(), buffer_infos.end(),
119 std::back_inserter(result), [](const BufferInfo& buffer_info) {
120 return buffer_info.is_temp_buffer();
121 });
122 return result;
123}
124
125// Add (from,to) rewrite pairs based on the given shape. These rewrite pairs
126// are used to generate methods for args and results.

Callers 1

GenerateHeaderFunction · 0.85

Calls 3

is_temp_bufferMethod · 0.80
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected