Home
last modified time | relevance | path

Searched refs:scalar_ty (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dhlo_utils.cc58 RankedTensorType scalar_ty = RankedTensorType::get({}, ty); in GetScalarOfType() local
62 return DenseElementsAttr::get(scalar_ty, value); in GetScalarOfType()
65 return DenseElementsAttr::get(scalar_ty, value); in GetScalarOfType()
70 scalar_ty, static_cast<std::complex<float>>(raw_value)); in GetScalarOfType()
73 scalar_ty, static_cast<std::complex<double>>(raw_value)); in GetScalarOfType()
124 RankedTensorType scalar_ty = RankedTensorType::get({}, ty); in GetScalarLimitOfType() local
126 return DenseElementsAttr::get(scalar_ty, in GetScalarLimitOfType()
130 scalar_ty, GetScalarLimitOfIntegerType(integer_ty, limit)); in GetScalarLimitOfType()
/external/llvm-project/llvm/utils/
Dshuffle_select_fuzz_tester.py376 scalar_ty = ty.get_scalar_type()
384 ty = ty.dump(), i_f = i_f, scalar_ty = scalar_ty.dump(),
387 check_die += die_template.format(lane = str(i), scalar_ty = scalar_ty.dump())
396 …inputs_values = [', '.join([scalar_ty.dump() + ' ' + str(i) for i in inp]) for inp in inputs_value…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Ddecompose_resource_ops.cc30 RankedTensorType scalar_ty = RankedTensorType::get({}, ty); in GetScalarOfType() local
33 return DenseElementsAttr::get(scalar_ty, attr); in GetScalarOfType()
38 return DenseElementsAttr::get(scalar_ty, attr); in GetScalarOfType()
Dlower_tf.cc75 RankedTensorType scalar_ty = RankedTensorType::get({}, ty); in GetScalarOfType() local
78 return DenseElementsAttr::get(scalar_ty, attr); in GetScalarOfType()
81 return DenseElementsAttr::get(scalar_ty, attr); in GetScalarOfType()
86 scalar_ty, static_cast<std::complex<float>>(raw_value)); in GetScalarOfType()
89 scalar_ty, static_cast<std::complex<double>>(raw_value)); in GetScalarOfType()
343 auto scalar_ty = RankedTensorType::get({}, element_ty); in matchAndRewrite() local
358 scalar_ty, ConvertToAPFloat(bits_min, element_ty))); in matchAndRewrite()
362 scalar_ty, ConvertToAPFloat(bits_max, element_ty))); in matchAndRewrite()
367 scalar_ty, ConvertToAPFloat(bits_max - bits_min, element_ty))); in matchAndRewrite()
383 rewriter.create<RoundOp>(op.getLoc(), scalar_ty, min_scaled_sub); in matchAndRewrite()
[all …]
/external/llvm/lib/Target/AMDGPU/
DR600Instructions.td1628 ValueType scalar_ty> : Pat <
1629 (scalar_ty (extractelt vec_ty:$vec, i32:$index)),
1639 ValueType scalar_ty> : Pat <
1640 (vec_ty (insertelt vec_ty:$vec, scalar_ty:$value, i32:$index)),
/external/llvm-project/llvm/lib/Target/AMDGPU/
DR600Instructions.td1700 ValueType scalar_ty> : R600Pat <
1701 (scalar_ty (extractelt vec_ty:$vec, i32:$index)),
1711 ValueType scalar_ty> : R600Pat <
1712 (vec_ty (insertelt vec_ty:$vec, scalar_ty:$value, i32:$index)),
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DR600Instructions.td1700 ValueType scalar_ty> : R600Pat <
1701 (scalar_ty (extractelt vec_ty:$vec, i32:$index)),
1711 ValueType scalar_ty> : R600Pat <
1712 (vec_ty (insertelt vec_ty:$vec, scalar_ty:$value, i32:$index)),
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc909 auto scalar_ty = RankedTensorType::get({}, element_ty); in GetEpsilonValue() local
913 return DenseElementsAttr::get(scalar_ty, value); in GetEpsilonValue()
917 return DenseElementsAttr::get(scalar_ty, value); in GetEpsilonValue()
920 return DenseElementsAttr::get(scalar_ty, value); in GetEpsilonValue()
923 return DenseElementsAttr::get(scalar_ty, value); in GetEpsilonValue()
2656 auto scalar_ty = RankedTensorType::get({}, operand_ty.getElementType()); in matchAndRewrite() local
2657 ElementsAttr attr = mlir::hlo::getSplat(&rewriter, scalar_ty, 0.5); in matchAndRewrite()