Home
last modified time | relevance | path

Searched refs:ConvertGatherDimensionNumbers (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.h33 mlir::mhlo::GatherDimensionNumbers ConvertGatherDimensionNumbers(
Dattribute_importer.cc48 mlir::mhlo::GatherDimensionNumbers ConvertGatherDimensionNumbers( in ConvertGatherDimensionNumbers() function
Dhlo_function_importer.cc350 ConvertGatherDimensionNumbers( in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc295 ConvertGatherDimensionNumbers(dimension_numbers, &builder_), in GatherInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1245 gather.dimension_numbersAttr(xla::ConvertGatherDimensionNumbers( in EmitGatherOp()
Dlegalize_tf.cc1083 return ::xla::ConvertGatherDimensionNumbers(dims, builder); in GetGatherDimNumsAttr()