Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.h35 mlir::mhlo::GatherDimensionNumbersAttr ConvertGatherDimensionNumbers(
Dattribute_importer.cc46 mlir::mhlo::GatherDimensionNumbersAttr ConvertGatherDimensionNumbers( in ConvertGatherDimensionNumbers() function
Dhlo_function_importer.cc695 ConvertGatherDimensionNumbers( in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc389 ConvertGatherDimensionNumbers(dimension_numbers, &builder_), in GatherInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc934 return ::xla::ConvertGatherDimensionNumbers(dims, builder); in GetGatherDimNumsAttr()