| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 95 HloInstruction* updates, absl::Span<const int64> update_window_dims) { in PermuteScatterAndWindowDims() argument
101 bool is_scatter_dim = !absl::c_binary_search(update_window_dims, i); in PermuteScatterAndWindowDims()
106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims()
388 updates, AsInt64Slice(dim_numbers.update_window_dims()))); in ExpandScatter()
|
| D | shape_inference.cc | 3073 if (!absl::c_is_sorted(dim_numbers.update_window_dims())) { in ValidateScatterDimensionNumbers()
3076 StrJoin(dim_numbers.update_window_dims(), ", ")); in ValidateScatterDimensionNumbers()
3078 if (absl::c_adjacent_find(dim_numbers.update_window_dims()) != in ValidateScatterDimensionNumbers()
3079 dim_numbers.update_window_dims().end()) { in ValidateScatterDimensionNumbers()
3082 StrJoin(dim_numbers.update_window_dims(), ", ")); in ValidateScatterDimensionNumbers()
3085 for (int64 window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers()
3226 auto update_window_dim = scatter_dim_numbers.update_window_dims(i); in InferScatterShape()
3241 absl::c_binary_search(scatter_dim_numbers.update_window_dims(), i); in InferScatterShape()
|
| D | hlo_instructions.cc | 2409 string update_window_dims = StrCat( in ScatterDimensionNumbersToString() local
2411 StrJoin(scatter_dimension_numbers().update_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2423 {update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims, in ScatterDimensionNumbersToString()
2430 absl::Span<const int64> update_window_dims, in MakeScatterDimNumbers() argument
2435 for (int64 update_window_dim : update_window_dims) { in MakeScatterDimNumbers()
|
| D | hlo_evaluator_typed_visitor.h | 2065 !absl::c_binary_search(dim_numbers.update_window_dims(), i); in IterationSpaceForUpdateScatterIndices()
2084 absl::c_binary_search(dim_numbers.update_window_dims(), i); in IterationSpaceForUpdateWindowIndices()
2111 !absl::c_binary_search(dim_numbers_.update_window_dims(), i)); in UpdateScatterIndexToInputIndex()
2241 if (absl::c_binary_search(dim_numbers.update_window_dims(), i)) { in UpdateWindowIndexToInputIndex()
|
| D | hlo_parser.cc | 1649 optional<std::vector<int64>> update_window_dims; in ParseInstructionRhs() local
1651 /*required=*/true, AttrTy::kBracedInt64List, &update_window_dims}; in ParseInstructionRhs()
1674 /*update_window_dims=*/*update_window_dims, in ParseInstructionRhs()
|
| D | hlo_instructions.h | 1417 absl::Span<const int64> update_window_dims,
|
| D | hlo_parser_test.cc | 872 …ndices, f32[10,9,8,7,30,29,28,27,26]{8,7,6,5,4,3,2,1,0} %updates), update_window_dims={4,5,6,7,8},… in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 195 << absl::StrJoin(dim_numbers.update_window_dims(), ",") << "]"; in XlaScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2115 …d, scatter_indices, updates, update_computation, index_vector_dim, update_window_dims, inserted_wi…
2124 `update_window_dims` | `ArraySlice<int64>` | The set of dimensions in `updates` shape tha…
2132 dimensions in `updates` shape that are not in `update_window_dims`, in ascending
2137 - `updates` array must be of rank `update_window_dims.size +
2142 - If `i` is present in `update_window_dims` (i.e. equal to
2143 `update_window_dims`[`k`] for some `k`), then the bound of dimension `i`
2156 - `update_window_dims` must be in ascending order, not have any repeating
2179 at `update_window_dims` in `U` according to `inserted_window_dims`. More
2182 < `update_window_dims.size`, where `window_dims_to_operand_dims` is the
2183 monotonic function with domain [`0`, `update_window_dims.size`) and
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1967 self.update_window_dims = []
|
| D | xla_client_test.py | 1684 dnums.update_window_dims.append(1)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 471 repeated int64 update_window_dims = 1; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1094 if (absl::c_binary_search(dim_numbers.update_window_dims(), i)) { in EmitScatter()
|