Home
last modified time | relevance | path

Searched defs:element_ty (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/
Dhlo_utils.h41 Type element_ty = getElementTypeOrSelf(ty); in getSplat() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape.cc55 mlir::Type element_ty = complex_type.getElementType(); in TypeToPrimitiveType() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc99 Type element_ty = shaped_ty.getElementType(); in ConstantFoldFallbackHook() local
Dlower_tf.cc342 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local
1035 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc377 Type element_ty = getElementTypeOrSelf(ty); in DropTypeHelper() local
Dtf_ops_n_z.cc149 Type element_ty = on_value.getType().cast<TensorType>().getElementType(); in InferOneHotOpType() local
581 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType() local
811 Type element_ty = e.getType().cast<TensorType>().getElementType(); in InferSelectV2OpType() local
Dtf_ops_a_m.cc2127 Type element_ty = input.getType().cast<TensorType>().getElementType(); in InferExpandDimsOpType() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc462 Type element_ty = operand_ty.getElementType(); in inferReturnTypes() local
964 auto element_ty = ComplexType::get(getElementTypeOrSelf(type)); in inferReturnTypes() local
993 auto element_ty = getElementTypeOrSelf(type); in CreateRealType() local
1752 Type element_ty = getElementTypeOrSelf(operand_ty); in GetReduceResultType() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc476 mlir::Type element_ty = getElementTypeOrSelf(type); in GetSplat() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1353 auto element_ty = input_ty.getElementType(); in GetReshapeOutputType() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc908 auto element_ty = ty.cast<TensorType>().getElementType(); in GetEpsilonValue() local