Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
H A Dhlo_ops_base.h142 if (auto rankedType = inputType.dyn_cast<RankedTensorType>()) in inferMostSpecificType() local
152 for (auto rankedType : rankedTypes) { in inferMostSpecificType() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DBase.h186 if (auto rankedType = inputType.dyn_cast<RankedTensorType>()) in inferMostSpecificType() local
197 for (auto rankedType : rankedTypes) { in inferMostSpecificType() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dhlo_legalize_to_memref.cc193 if (auto rankedType = opResultType.dyn_cast<RankedTensorType>()) { in bufferize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/
H A Dshape_component_analysis.cc704 if (auto rankedType = type.dyn_cast<RankedTensorType>()) in dim0size() local