Home
last modified time | relevance | path

Searched defs:dimsType (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dunfuse_batch_norm.cc42 auto dimsType = RankedTensorType::get({1}, rewriter.getIntegerType(64)); in broadcastToFeatureDim() local
89 auto dimsType = RankedTensorType::get({0}, b.getIntegerType(64)); in materializeEpsilon() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc5930 auto dimsType = RankedTensorType::get({0}, rewriter.getIntegerType(64)); in matchAndRewrite() local
5943 auto dimsType = RankedTensorType::get({0}, rewriter.getIntegerType(64)); in matchAndRewrite() local
5985 auto dimsType = RankedTensorType::get({0}, rewriter.getIntegerType(64)); in matchAndRewrite() local
/aosp_15_r20/external/skia/src/sksl/codegen/
H A DSkSLSPIRVCodeGenerator.cpp2295 SpvId dimsType = this->getType(*fContext.fTypes.fUInt2); in writeSpecialIntrinsic() local