Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dattribute_importer.cc65 std::vector<int64_t> scatter_dims_to_operand_dims( in ConvertScatterDimensionNumbers() local
H A Dmlir_hlo_to_hlo.cc446 auto scatter_dims_to_operand_dims = input.getScatterDimsToOperandDims(); in Convert_scatter_dimension_numbers() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1138 std::vector<int64_t> scatter_dims_to_operand_dims( in ScatterOutputShardingFromUpdate() local
1157 std::vector<int64_t> scatter_dims_to_operand_dims( in ScatterUpdateShardingFromOutput() local
H A Dhlo_instructions.cc3165 std::string scatter_dims_to_operand_dims = StrCat( in ScatterDimensionNumbersToString() local
3182 absl::Span<const int64_t> scatter_dims_to_operand_dims, in MakeScatterDimNumbers()
H A Dhlo_parser.cc2853 optional<std::vector<int64_t>> scatter_dims_to_operand_dims; in CreateInstruction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto558 repeated int64 scatter_dims_to_operand_dims = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc3177 auto scatter_dims_to_operand_dims = in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc5739 SmallVector<int64_t, 1> scatter_dims_to_operand_dims(1, 0); in matchAndRewrite() local