Home
last modified time | relevance | path

Searched refs:GetScalarOfType (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.td21 class GetScalarOfType<int value> : NativeCodeCall<
22 "GetScalarOfType(getElementTypeOrSelf($0)," # value # ")">;
25 "GetScalarOfType(getElementTypeOrSelf($0)," # value # ")">;
28 "GetScalarOfType(getElementTypeOrSelf($0), "
32 "GetScalarOfType(getElementTypeOrSelf($0), "
97 "GetScalarOfType(getElementTypeOrSelf($1), "
102 "GetScalarOfType($_builder.getIntegerType(32), "
117 (TF_ConstOp (GetScalarOfType<1> $features)),
118 (TF_ConstOp (GetScalarOfType<0> $features)),
123 (TF_ConstOp (GetScalarOfType<0> $sparse_labels)), $sparse_labels),
[all …]
Ddecompose_resource_ops.td21 class GetScalarOfType<int value> : NativeCodeCall<
22 "GetScalarOfType(getElementTypeOrSelf($0)," # value # ")">;
214 (TF_ConstOp:$zero_epsilon (GetScalarOfType<0> $grad)),
235 (TF_ConstOp:$one (GetScalarOfType<1> $grad)),
281 (TF_ConstOp:$one (GetScalarOfType<1> $grad)),
362 [(TF_ConstOp:$one (GetScalarOfType<1> $grad)),
424 [(TF_ConstOp:$one (GetScalarOfType<1> $grad)),
461 [(TF_ConstOp:$one (GetScalarOfType<1> $grad)),
462 (TF_ConstOp:$zero (GetScalarOfType<0> $grad)),
Ddecompose_resource_ops.cc29 static DenseElementsAttr GetScalarOfType(Type ty, int64_t raw_value) { in GetScalarOfType() function
168 Value step_size = rewriter.create<ConstOp>(loc, GetScalarOfType(u64, 256)); in matchAndRewrite()
175 Value zero_u64 = rewriter.create<ConstOp>(loc, GetScalarOfType(u64, 0)); in matchAndRewrite()
176 Value one_u64 = rewriter.create<ConstOp>(loc, GetScalarOfType(u64, 1)); in matchAndRewrite()
Dlower_tf.cc74 static DenseElementsAttr GetScalarOfType(Type ty, T raw_value) { in GetScalarOfType() function
136 loc, GetScalarOfType(rewriter.getIntegerType(64), 0)); in ValuesToRank1()
477 auto start = rewriter.create<ConstOp>(loc, GetScalarOfType(int_type, 0)); in matchAndRewrite()
479 loc, GetScalarOfType(int_type, x_type.getShape()[0])); in matchAndRewrite()
480 auto delta = rewriter.create<ConstOp>(loc, GetScalarOfType(int_type, 1)); in matchAndRewrite()
865 loc, GetScalarOfType(rewriter.getIntegerType(64), 0)); in matchAndRewrite()
921 loc, GetScalarOfType(rewriter.getIntegerType(32), 0)); in matchAndRewrite()
1359 rewriter.create<ConstOp>(loc, GetScalarOfType(out_size_element_ty, 1)); in matchAndRewrite()
1408 loc, GetScalarOfType(rewriter.getF32Type(), 0.0)); in matchAndRewrite()
1410 loc, GetScalarOfType(rewriter.getF32Type(), 1.0)); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_patterns.td146 (HLOClient_BroadcastCompareOp $l, (HLO_ConstOp (GetScalarOfType<0> $l)),
149 (HLOClient_BroadcastCompareOp $r, (HLO_ConstOp (GetScalarOfType<0> $r)),
158 (HLO_ConstOp (GetScalarOfType<1> $r)),
176 (HLO_ConstOp:$l_zeros (GetScalarOfType<0> $l)),
181 (HLO_ConstOp:$r_zeros (GetScalarOfType<0> $r)),
469 (HLO_ConstOp:$zero (GetScalarOfType<0> $features)),
479 (HLO_ConstOp:$zero (GetScalarOfType<0> $features)),
487 (HLO_ConstOp:$one (GetScalarOfType<1> $features)),
498 // them in GetScalarOfType.
501 (HLO_ConstOp:$zero (GetScalarOfType<0> $input)), $input,
[all …]
Dlegalize_tf.cc276 return builder->create<ConstOp>(loc, hlo::GetScalarOfType(ty, raw_value)); in GetScalarConstOfType()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_utils.td52 class GetScalarOfType<int value> : NativeCodeCall<
53 "hlo::GetScalarOfType(getElementTypeOrSelf($0)," # value # ")">;
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/
Dhlo_utils.h69 DenseElementsAttr GetScalarOfType(Type ty, int64_t raw_value);
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_canonicalize.td29 (HLO_EinsumOp (HLO_ConstOp (GetScalarOfType<1> $operand)),
Dmhlo_canonicalize.td29 (HLO_EinsumOp (HLO_ConstOp (GetScalarOfType<1> $operand)),
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dhlo_utils.cc57 DenseElementsAttr GetScalarOfType(Type ty, int64_t raw_value) { in GetScalarOfType() function