| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 223 scatter->scatter_dimension_numbers(); in ScatterLoopBody()
334 scatter->scatter_dimension_numbers(); in ScatterTripCount()
367 scatter->scatter_dimension_numbers(); in ExpandInstruction()
|
| D | hlo_sharding_util.cc | 645 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterIndexSharding()
678 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterDataSharding()
713 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveIndexSharding()
756 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveDataSharding()
1009 const auto& dnums = hlo.scatter_dimension_numbers(); in ScatterOutputShardingFromUpdate()
1033 const auto& dnums = hlo.scatter_dimension_numbers(); in ScatterUpdateShardingFromOutput()
|
| D | hlo_instructions.cc | 2834 const ScatterDimensionNumbers& scatter_dimension_numbers) { in ScatterDimensionNumbersToString() argument
2837 StrJoin(scatter_dimension_numbers.update_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2840 StrJoin(scatter_dimension_numbers.inserted_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2843 StrJoin(scatter_dimension_numbers.scatter_dims_to_operand_dims(), ","), in ScatterDimensionNumbersToString()
2846 StrCat("index_vector_dim=", scatter_dimension_numbers.index_vector_dim()); in ScatterDimensionNumbersToString()
2877 *proto.mutable_scatter_dimension_numbers() = scatter_dimension_numbers(); in ToProto()
2886 ScatterDimensionNumbersToString(scatter_dimension_numbers())}; in ExtraAttributesToStringImpl()
2902 scatter_dimension_numbers(), in IdenticalSlowPath()
2903 casted_other.scatter_dimension_numbers()) && in IdenticalSlowPath()
2915 scatter_dimension_numbers(), indices_are_sorted(), unique_indices()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 685 auto scatter_dimension_numbers = in CreateFromProto() local
687 proto.scatter_dimension_numbers()); in CreateFromProto()
690 computations(0), *scatter_dimension_numbers, in CreateFromProto()
4244 const ScatterDimensionNumbers& HloInstruction::scatter_dimension_numbers() in scatter_dimension_numbers() function in xla::HloInstruction
4246 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
|
| D | hlo_instructions.h | 1706 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers() function
1726 const ScatterDimensionNumbers& scatter_dimension_numbers);
|
| D | hlo.proto | 194 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48; field
|
| D | hlo_evaluator_typed_visitor.h | 2332 scatter->scatter_dimension_numbers();
2355 &scatter->scatter_dimension_numbers(), /*input_shape=*/operand_shape,
2358 scatter->scatter_dimension_numbers(), /*input_shape=*/operand_shape,
|
| D | hlo_instruction.h | 1888 const ScatterDimensionNumbers& scatter_dimension_numbers() const;
|
| D | dynamic_dimension_inference.cc | 1415 hlo->scatter_dimension_numbers(); in HandleScatter()
|
| D | hlo_verifier.cc | 1145 scatter->scatter_dimension_numbers())); in HandleScatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | gpu_ops.mlir | 25 }) { scatter_dimension_numbers = {index_vector_dim = 1 : i64,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | canonicalize.mlir | 1253 scatter_dimension_numbers = {
1276 scatter_dimension_numbers = {
1300 scatter_dimension_numbers = {
1324 scatter_dimension_numbers = {
1348 scatter_dimension_numbers = {
1371 scatter_dimension_numbers = {
1396 scatter_dimension_numbers = {
1421 scatter_dimension_numbers = {
1445 scatter_dimension_numbers = {
1466 scatter_dimension_numbers = {
[all …]
|
| D | lhlo_ops.mlir | 910 scatter_dimension_numbers = {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1109 scatter_op.scatter_dimension_numbers().index_vector_dim().getInt(); in matchAndRewrite()
1119 scatter_op.scatter_dimension_numbers().inserted_window_dims(); in matchAndRewrite()
1121 scatter_op.scatter_dimension_numbers().scatter_dims_to_operand_dims(); in matchAndRewrite()
1136 scatter_op.scatter_dimension_numbers().update_window_dims(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 548 scatter_instr->scatter_dimension_numbers(); in GetScatterDimensionNumbers()
549 auto scatter_dimension_numbers = mhlo::ScatterDimensionNumbers::get( in GetScatterDimensionNumbers() local
555 return scatter_dimension_numbers; in GetScatterDimensionNumbers()
566 TF_ASSIGN_OR_RETURN(auto scatter_dimension_numbers, in EmitScatterOp()
568 scatter.scatter_dimension_numbersAttr(scatter_dimension_numbers); in EmitScatterOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 2999 scatter_dimension_numbers().index_vector_dim().getInt(); in fold()
3036 if (llvm::count(scatter_dimension_numbers().update_window_dims(), i) == 0) in fold()
3045 uint64_t operand_dim = scatter_dimension_numbers() in fold()
3055 if (llvm::count(scatter_dimension_numbers().inserted_window_dims(), i)) in fold()
3058 update_index[scatter_dimension_numbers() in fold()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 585 : use->scatter_dimension_numbers() in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 509 auto dnums = scatter->scatter_dimension_numbers(); in HandleScatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-with-tf2xla.mlir | 204 // CHECK-SAME: scatter_dimension_numbers
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | legalize_hlo.mlir | 1948 scatter_dimension_numbers = {
1971 scatter_dimension_numbers = {
1994 scatter_dimension_numbers = {
2017 scatter_dimension_numbers = {
2040 scatter_dimension_numbers = {
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 472 ScatterDimensionNumbers:$scatter_dimension_numbers,
|
| D | hlo_ops.td | 1092 ScatterDimensionNumbers:$scatter_dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 434 ConvertScatterDimensionNumbers(scatter->scatter_dimension_numbers(), in ImportInstructionImpl()
|
| D | mlir_hlo_to_hlo.cc | 948 Convert_scatter_dimension_numbers(op.scatter_dimension_numbers()); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 763 scatter_dimension_numbers = {
|