- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ConvertToTfliteSize (0.24 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
SmallVector<Attribute, 4> result_shape_data(4); for (int i = 0; i < 4; ++i) { auto size = result_shape[i]; result_shape_data[i] = rewriter.getI32IntegerAttr(ConvertToTfliteSize(size)); } auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); auto shape = rewriter.create<TF::ConstOp>(loc, shape_type, shape_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
mlir::SmallVector<mlir::Attribute, 4> shape; for (auto s : new_shape) { shape.push_back( builder.getI32IntegerAttr(mlir::TFL::ConvertToTfliteSize(s))); } auto output_shape = DenseElementsAttr::get(shape_type, shape); auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); op_state.addOperands({shape_op}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)