Home
last modified time | relevance | path

Searched refs:inserted_window_dims (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc197 << absl::StrJoin(dim_numbers.inserted_window_dims(), ",") << "]"; in XlaScatter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3095 if (!absl::c_is_sorted(dim_numbers.inserted_window_dims())) { in ValidateScatterDimensionNumbers()
3098 StrJoin(dim_numbers.inserted_window_dims(), ", ")); in ValidateScatterDimensionNumbers()
3100 if (absl::c_adjacent_find(dim_numbers.inserted_window_dims()) != in ValidateScatterDimensionNumbers()
3101 dim_numbers.inserted_window_dims().end()) { in ValidateScatterDimensionNumbers()
3104 StrJoin(dim_numbers.inserted_window_dims(), ", ")); in ValidateScatterDimensionNumbers()
3106 for (int64 inserted_dim : dim_numbers.inserted_window_dims()) { in ValidateScatterDimensionNumbers()
3219 scatter_dim_numbers.inserted_window_dims(inserted_dims_seen) == i) { in InferScatterShape()
Dscatter_expander.cc279 AsInt64Slice(dim_numbers.inserted_window_dims()))); in ScatterLoopBody()
Dhlo_instructions.cc2412 string inserted_window_dims = StrCat( in ScatterDimensionNumbersToString() local
2414 StrJoin(scatter_dimension_numbers().inserted_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2423 {update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims, in ScatterDimensionNumbersToString()
2431 absl::Span<const int64> inserted_window_dims, in MakeScatterDimNumbers() argument
2438 for (int64 inserted_window_dim : inserted_window_dims) { in MakeScatterDimNumbers()
Dhlo_parser.cc1652 optional<std::vector<int64>> inserted_window_dims; in ParseInstructionRhs() local
1654 /*required=*/true, AttrTy::kBracedInt64List, &inserted_window_dims}; in ParseInstructionRhs()
1675 /*inserted_window_dims=*/*inserted_window_dims, in ParseInstructionRhs()
Dhlo_instructions.h1418 absl::Span<const int64> inserted_window_dims,
Dhlo_parser_test.cc872 …26]{8,7,6,5,4,3,2,1,0} %updates), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter… in CreateTestCases()
Dhlo_evaluator_typed_visitor.h2250 if (absl::c_binary_search(dim_numbers.inserted_window_dims(), i)) { in UpdateWindowIndexToInputIndex()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2115 …pdates, update_computation, index_vector_dim, update_window_dims, inserted_window_dims, scatter_di…
2125 `inserted_window_dims` | `ArraySlice<int64>` | The set of _window dimensions_ that must be …
2145 accounting for the `inserted_window_dims` (i.e.
2148 `inserted_window_dims` removed).
2159 - `inserted_window_dims` must be in ascending order, not have any repeating
2179 at `update_window_dims` in `U` according to `inserted_window_dims`. More
2184 range [`0`, `operand.rank`) \\ `inserted_window_dims`. (For example, if
2186 `inserted_window_dims` is {`0`, `2`} then `window_dims_to_operand_dims`
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1968 self.inserted_window_dims = []
Dxla_client_test.py1685 dnums.inserted_window_dims.append(0)
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto473 repeated int64 inserted_window_dims = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1109 if (absl::c_binary_search(dim_numbers.inserted_window_dims(), i)) { in EmitScatter()