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