Home
last modified time | relevance | path

Searched defs:scatter_dims_to_operand_dims (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc69 std::vector<int64_t> scatter_dims_to_operand_dims( in ConvertScatterDimensionNumbers() local
Dmlir_hlo_to_hlo.cc355 auto scatter_dims_to_operand_dims = in Convert_scatter_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc541 std::vector<int64> scatter_dims_to_operand_dims( in HandleScatter() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1055 std::vector<int64> scatter_dims_to_operand_dims( in ScatterOutputShardingFromUpdate() local
1079 std::vector<int64> scatter_dims_to_operand_dims( in ScatterUpdateShardingFromOutput() local
Dhlo_instructions.cc3005 string scatter_dims_to_operand_dims = StrCat( in ScatterDimensionNumbersToString() local
3022 absl::Span<const int64> scatter_dims_to_operand_dims, in MakeScatterDimNumbers()
Dhlo_parser.cc2648 optional<std::vector<int64>> scatter_dims_to_operand_dims; in ParseInstructionRhs() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto516 repeated int64 scatter_dims_to_operand_dims = 3; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1795 auto scatter_dims_to_operand_dims = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2634 auto scatter_dims_to_operand_dims = Extract1DVector( in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6017 SmallVector<int64_t, 1> scatter_dims_to_operand_dims(1, 0); in matchAndRewrite() local