- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for replaceMap (0.19 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
} auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp( &state, rewriter, op->getLoc(), tf_op.getType(), input, filter, bias); rewriter.replaceOp(op, conv_op.getResult()); return success(); } const IntegerAttr intAttrOne; private: bool allow_bf16_and_f16_type_legalization_; };
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/tensorflow/transforms/lower_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
if ((ranked_type && ranked_type.getRank() == 0) || !matchPattern(element_shape, m_Constant(&dense_elem_attr))) { // If no constant is spotted, just forward the operand. rewriter.replaceOp(op, {input}); return success(); } RankedTensorType shape_type = tensorflow::GetTypeFromTFTensorShape({-1}, rewriter.getIntegerType(32));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
} auto scast_op = rewriter.create<quantfork::StorageCastOp>( call_op.getLoc(), result_type, result); replace_map.insert(std::make_pair(result, scast_op)); } for (auto replace_pair : replace_map) { Value result = replace_pair.first; quantfork::StorageCastOp scast_op = replace_pair.second; result.replaceAllUsesExcept(scast_op, scast_op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)