Searched refs:GetScalarConstOfType (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 274 static ConstOp GetScalarConstOfType(Type ty, Location loc, int64_t raw_value, in GetScalarConstOfType() function
315 GetScalarConstOfType(getElementTypeOrSelf(minor_indices[0].getType()), in CreateFullIndexVectorFromMinorIndices()
364 dus_starts[i] = GetScalarConstOfType(builder->getIntegerType(32), loc, in UpdateSliceInMinorDims()
1403 Value zero = GetScalarConstOfType(input_type.getElementType(), op.getLoc(), in matchAndRewrite()
1457 GetScalarConstOfType(rewriter.getIntegerType(int_size), loc, constant, in BroadcastConstant()
1780 GetScalarConstOfType(input_ty.getElementType(), loc, 0, &rewriter); in matchAndRewrite()
2212 GetScalarConstOfType(element_type, loc, window_count, &rewriter); in AvgPoolDivideByCount()
2286 GetScalarConstOfType(sum_element_type, op.getLoc(), 0, &rewriter); in matchAndRewrite()
2398 Value zero = GetScalarConstOfType(element_type, loc, 0, &rewriter); in matchAndRewrite()
2469 zero = GetScalarConstOfType(sum_element_type, loc, 0, &rewriter); in matchAndRewrite()
[all …]
|