/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
H A D | ChloOps.cpp | 190 DictionaryAttr attributes, RegionRange /*regions*/, in inferReturnTypeComponents() argument 225 DictionaryAttr attributes, RegionRange /*regions*/, in inferReturnTypeComponents() argument 252 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 263 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 274 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 349 RegionRange /*regions*/, in inferReturnTypeComponents() argument 382 DictionaryAttr, RegionRange, in inferReturnTypeComponents() argument 500 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument
|
H A D | StablehloOps.cpp | 758 DictionaryAttr attributes, RegionRange, in inferReturnTypes() argument 1014 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 1805 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 2866 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 2890 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 2902 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 2916 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 4192 DictionaryAttr attributes, RegionRange, in inferReturnTypeComponents() argument 4565 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 4948 DictionaryAttr attributes, RegionRange, in inferReturnTypes() argument [all …]
|
H A D | Base.h | 138 RegionRange /*regions*/, SmallVectorImpl<Type> &inferredReturnTypes) { in inferReturnTypes() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
H A D | chlo_ops.cc | 217 DictionaryAttr attributes, RegionRange /*regions*/, in inferReturnTypeComponents() argument 252 DictionaryAttr attributes, RegionRange /*regions*/, in inferReturnTypeComponents() argument 279 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 290 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 301 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 376 RegionRange /*regions*/, in inferReturnTypeComponents() argument 409 DictionaryAttr, RegionRange, in inferReturnTypeComponents() argument
|
H A D | hlo_ops.cc | 756 DictionaryAttr attributes, RegionRange, in inferReturnTypes() argument 1012 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 2128 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 3579 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 3612 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 3643 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 3656 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 5505 RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument 5583 DictionaryAttr attributes, RegionRange, in inferReturnTypeComponents() argument 6161 DictionaryAttr, RegionRange, SmallVectorImpl<Type>& inferredReturnTypes) { in inferReturnTypes() argument [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
H A D | hlo_ops_base.h | 92 DictionaryAttr /*attributes*/, RegionRange /*regions*/, in inferReturnTypes() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
H A D | tfl_ops.cc | 1205 DictionaryAttr attr, RegionRange, in inferReturnTypes() argument 1880 DictionaryAttr attr, RegionRange, in inferReturnTypes() argument 2608 RegionRange, SmallVectorImpl<Type> &inferredReturnTypes) { in inferReturnTypes() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/ |
H A D | gml_st_ops.cc | 141 DictionaryAttr attributes, RegionRange, in inferReturnTypes() argument
|