Searched refs:add_layout (Results 1 – 2 of 2) sorted by relevance
473 tensor_builder.add_layout(ToFB(desc.layout)); in Encode()
1440 void add_layout(tflite::gpu::data::Layout layout) { in add_layout() function1472 builder_.add_layout(layout);