Searched refs:bn_train_input (Results 1 – 1 of 1) sorted by relevance
1959 Value bn_train_input = rewriter.create<mhlo::ConvertOp>(op.getLoc(), op.x(), in matchAndRewrite() local1962 bn_train_input.getType().template cast<TensorType>(); in matchAndRewrite()1980 op.getLoc(), result_type, bn_train_input, op.scale(), op.offset(), in matchAndRewrite()2078 /*result_type=*/bn_train_input_type_tensor, bn_train_input, in matchAndRewrite()