Home
last modified time | relevance | path

Searched defs:input_element_type (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize_op_order.cc76 Type input_element_type = getElementTypeOrSelf(dequantize_op.input()); in matchAndRewrite() local
H A Dlower_static_tensor_list.cc981 Type input_element_type = getElementTypeOrSelf(input); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc807 static void BuildArgMinMaxReductionBody(Type input_element_type, in BuildArgMinMaxReductionBody()
2348 auto input_element_type = input_type_tensor.getElementType(); in matchAndRewrite() local
2666 Type input_element_type = input_type.getElementType(); in matchAndRewrite() local
4344 Type input_element_type = input_type.getElementType(); in matchAndRewrite() local
6305 Type input_element_type = input_type.getElementType(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc3250 auto input_element_type = input_type.getElementType(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tfl.cc2993 UniformQuantizedType input_element_type = in matchAndRewrite() local