Home
last modified time | relevance | path

Searched defs:scalar_ty (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Drewrite_util.h29 RankedTensorType scalar_ty = RankedTensorType::get({}, ty); in GetScalarOfType() local
H A Dlower_tf.cc377 auto scalar_ty = RankedTensorType::get({}, element_ty); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dregister_ops_common_utils.cpp20 at::ScalarType scalar_ty, in tensorToListRecursive()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc782 auto scalar_ty = RankedTensorType::get({}, element_ty); in GetEpsilonValue() local
3055 auto scalar_ty = RankedTensorType::get({}, operand_ty.getElementType()); in matchAndRewrite() local