Home
last modified time | relevance | path

Searched refs:scatter_dims_to_operand_dims (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc69 std::vector<int64_t> scatter_dims_to_operand_dims( in ConvertScatterDimensionNumbers() local
70 dnums.scatter_dims_to_operand_dims().begin(), in ConvertScatterDimensionNumbers()
71 dnums.scatter_dims_to_operand_dims().end()); in ConvertScatterDimensionNumbers()
75 Convert(scatter_dims_to_operand_dims, builder), in ConvertScatterDimensionNumbers()
Dmlir_hlo_to_hlo.cc331 auto scatter_dims_to_operand_dims = in Convert_scatter_dimension_numbers() local
332 ConvertDenseIntAttr(input.scatter_dims_to_operand_dims()); in Convert_scatter_dimension_numbers()
333 std::copy(scatter_dims_to_operand_dims.begin(), in Convert_scatter_dimension_numbers()
334 scatter_dims_to_operand_dims.end(), in Convert_scatter_dimension_numbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dgpu_ops.mlir16 // CHECK: scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>
27 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
Dhlo_text_to_lhlo_no_opt.hlotxt34 // CHECK: scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>
46 scatter_dims_to_operand_dims={0},
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc522 std::vector<int64> scatter_dims_to_operand_dims( in HandleScatter() local
523 dnums.scatter_dims_to_operand_dims().begin(), in HandleScatter()
524 dnums.scatter_dims_to_operand_dims().end()); in HandleScatter()
636 operand, scatter_dims_to_operand_dims, slice_size) && in HandleScatter()
650 operand, indices, partition_id_, scatter_dims_to_operand_dims, in HandleScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dscatter.hlo66 scatter_dims_to_operand_dims={0},
122 scatter_dims_to_operand_dims={},
207 scatter_dims_to_operand_dims={0},
268 scatter_dims_to_operand_dims={0},
Dfused_scatter.hlo183 scatter_dims_to_operand_dims={0},
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1012 std::vector<int64> scatter_dims_to_operand_dims( in ScatterOutputShardingFromUpdate() local
1013 dnums.scatter_dims_to_operand_dims().begin(), in ScatterOutputShardingFromUpdate()
1014 dnums.scatter_dims_to_operand_dims().end()); in ScatterOutputShardingFromUpdate()
1028 scatter_dims_to_operand_dims, update_window_dims, slice_size); in ScatterOutputShardingFromUpdate()
1036 std::vector<int64> scatter_dims_to_operand_dims( in ScatterUpdateShardingFromOutput() local
1037 dnums.scatter_dims_to_operand_dims().begin(), in ScatterUpdateShardingFromOutput()
1038 dnums.scatter_dims_to_operand_dims().end()); in ScatterUpdateShardingFromOutput()
1052 inserted_window_dims, scatter_dims_to_operand_dims, update_window_dims, in ScatterUpdateShardingFromOutput()
Dscatter_expander.cc154 FindIndex(dim_numbers.scatter_dims_to_operand_dims(), i); in ExpandIndexVectorIntoOperandSpace()
Dhlo_instructions.cc2841 string scatter_dims_to_operand_dims = StrCat( in ScatterDimensionNumbersToString() local
2843 StrJoin(scatter_dimension_numbers.scatter_dims_to_operand_dims(), ","), in ScatterDimensionNumbersToString()
2849 {update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims, in ScatterDimensionNumbersToString()
2858 absl::Span<const int64> scatter_dims_to_operand_dims, in MakeScatterDimNumbers() argument
2867 for (int64 scatter_dim_to_operand_dim : scatter_dims_to_operand_dims) { in MakeScatterDimNumbers()
Dhlo_parser_test.cc945 …es), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter_dims_to_operand_dims={0,1,2,… in CreateTestCases()
964 …es), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter_dims_to_operand_dims={0,1,2,… in CreateTestCases()
983 …es), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter_dims_to_operand_dims={0,1,2,… in CreateTestCases()
Dshape_inference.cc3512 dim_numbers.scatter_dims_to_operand_dims(i); in ValidateScatterDimensionNumbers()
3522 dim_numbers.scatter_dims_to_operand_dims().begin(), in ValidateScatterDimensionNumbers()
3523 dim_numbers.scatter_dims_to_operand_dims().end()); in ValidateScatterDimensionNumbers()
3530 StrJoin(dim_numbers.scatter_dims_to_operand_dims(), ", ")); in ValidateScatterDimensionNumbers()
Dhlo_parser.cc2492 optional<std::vector<int64>> scatter_dims_to_operand_dims; in ParseInstructionRhs() local
2495 &scatter_dims_to_operand_dims}; in ParseInstructionRhs()
2519 /*scatter_dims_to_operand_dims=*/*scatter_dims_to_operand_dims, in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td36 StructFieldAttr<"scatter_dims_to_operand_dims", I64ElementsAttr>,
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc199 << absl::StrJoin(dim_numbers.scatter_dims_to_operand_dims(), ",") in XlaScatter()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dcanonicalize.mlir1256 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1279 scatter_dims_to_operand_dims = dense<1> : tensor<1xi64>,
1303 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1327 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1351 scatter_dims_to_operand_dims = dense<1> : tensor<1xi64>,
1374 scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>,
1399 scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>,
1424 scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>,
1448 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1469 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
[all …]
Dlhlo_ops.mlir913 scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1120 auto scatter_dims_to_operand_dims = in matchAndRewrite() local
1121 scatter_op.scatter_dimension_numbers().scatter_dims_to_operand_dims(); in matchAndRewrite()
1123 !IsIotaAttr(scatter_dims_to_operand_dims, in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py659 self.scatter_dims_to_operand_dims = []
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc586 .scatter_dims_to_operand_dims(); in CreateLiteralForConstrainedUses()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir207 // CHECK-SAME: scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1951 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1974 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
1997 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
2020 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
2043 scatter_dims_to_operand_dims = dense<0> : tensor<1xi64>,
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2393 …on, index_vector_dim, update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims)` </b>
2404 …scatter_dims_to_operand_dims` | `ArraySlice<int64>` | A dimensions map from the scatter indices to…
2444 - `scatter_dims_to_operand_dims.size` must be equal to
2456 `S` using the `scatter_dims_to_operand_dims` map. More formally:
2457 1. `S`<sub>`in`</sub>[`scatter_dims_to_operand_dims`[`k`]] = `S`[`k`] if
2458 `k` < `scatter_dims_to_operand_dims.size`.
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir766 scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>,
781 …VAL_3]]), update_window_dims={1}, inserted_window_dims={0,1}, scatter_dims_to_operand_dims={0,1}, …
Dimport.hlotxt768 …updates), update_window_dims={1}, inserted_window_dims={0,1}, scatter_dims_to_operand_dims={0,1}, …
782 // CHECK-SAME: scatter_dims_to_operand_dims = dense<[0, 1]> : tensor<2xi64>

12