Home
last modified time | relevance | path

Searched defs:initValueTypes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp411 ArrayRef<TensorType> initValueTypes, int64_t numInputs, in verifyReducerShape()
3968 ArrayRef<TensorType> initValueTypes(operandTensorTypes.begin() + numInputs, in verify() local
5388 SmallVector<TensorType> inputTypes, initValueTypes; in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc468 ArrayRef<TensorType> initValueTypes, int64_t numInputs, in verifyReducerShape()
5232 ArrayRef<TensorType> initValueTypes(operandTensorTypes.begin() + numInputs, in verify() local
7970 SmallVector<TensorType> inputTypes, initValueTypes; in verify() local