Home
last modified time | relevance | path

Searched refs:collapsed_slice_dims (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc207 AsInt64Slice(dim_numbers.collapsed_slice_dims()))); in GatherLoopBody()
240 if (!absl::c_binary_search(dim_numbers.collapsed_slice_dims(), i)) { in CreateGatherLoopAccumulatorInitValue()
Dshape_inference.cc2931 for (int64 collapsed_dim : dim_numbers.collapsed_slice_dims()) { in ValidateGatherDimensionNumbers()
2940 if (!absl::c_is_sorted(dim_numbers.collapsed_slice_dims())) { in ValidateGatherDimensionNumbers()
2943 StrJoin(dim_numbers.collapsed_slice_dims(), ", ")); in ValidateGatherDimensionNumbers()
2946 if (absl::c_adjacent_find(dim_numbers.collapsed_slice_dims()) != in ValidateGatherDimensionNumbers()
2947 dim_numbers.collapsed_slice_dims().end()) { in ValidateGatherDimensionNumbers()
2951 StrJoin(dim_numbers.collapsed_slice_dims(), ", ")); in ValidateGatherDimensionNumbers()
3014 StrJoin(gather_dim_numbers.collapsed_slice_dims(), ",")); in InferGatherShape()
3029 if (slice_sizes[gather_dim_numbers.collapsed_slice_dims(i)] != 1) { in InferGatherShape()
3033 slice_sizes[gather_dim_numbers.collapsed_slice_dims(i)], in InferGatherShape()
3034 gather_dim_numbers.collapsed_slice_dims(i), i); in InferGatherShape()
[all …]
Dindexed_array_analysis.cc267 dim_numbers.collapsed_slice_dims(0) != dim_numbers.start_index_map(0)) { in ComputeArrayForGather()
279 if (i != dim_numbers.collapsed_slice_dims(0) && in ComputeArrayForGather()
285 << dim_numbers.collapsed_slice_dims(0); in ComputeArrayForGather()
Dhlo_instructions.cc2326 string collapsed_slice_dims = StrCat( in GatherDimensionNumbersToString() local
2328 StrJoin(gather_dimension_numbers_->collapsed_slice_dims(), ","), "}"); in GatherDimensionNumbersToString()
2336 {offset_dims, collapsed_slice_dims, start_index_map, index_vector_dim}, in GatherDimensionNumbersToString()
2342 absl::Span<const int64> collapsed_slice_dims, in MakeGatherDimNumbers() argument
2348 for (int64 elided_window_dim : collapsed_slice_dims) { in MakeGatherDimNumbers()
Dhlo_evaluator.cc809 while (absl::c_binary_search(dim_numbers.collapsed_slice_dims(), in IterationSpaceForOutputOffsetIndices()
972 if (absl::c_binary_search(dim_numbers.collapsed_slice_dims(), i)) { in OutputOffsetIndexToInputIndex()
Dhlo_parser_test.cc853 …[10,9,8,7,5]{4,3,2,1,0} %start_indices), offset_dims={4,5,6,7,8}, collapsed_slice_dims={}, start_i… in CreateTestCases()
1333 …,0} gather(input_tensor, start_indices), offset_dims={4,5,6,7,8}, collapsed_slice_dims={}, start_i… in CreateTestCases()
Dhlo_parser.cc1618 optional<std::vector<int64>> collapsed_slice_dims; in ParseInstructionRhs() local
1620 /*required=*/true, AttrTy::kBracedInt64List, &collapsed_slice_dims}; in ParseInstructionRhs()
1639 /*collapsed_slice_dims=*/*collapsed_slice_dims, in ParseInstructionRhs()
Dhlo_instructions.h1381 absl::Span<const int64> collapsed_slice_dims,
Delemental_ir_emitter.cc1885 if (absl::c_binary_search(dim_numbers.collapsed_slice_dims(), i)) { in EmitElementalGather()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1269 <b> `gather(operand, start_indices, offset_dims, collapsed_slice_dims, slice_sizes, start_index_map…
1289 | `collapsed_slice_dims` | `ArraySlice<int64>` | The set of dimensions in each |
1302 `collapsed_slice_dims`.size.
1319 accounting for `collapsed_slice_dims` (i.e. we pick
1321 with the bounds at indices `collapsed_slice_dims` removed).
1339 at the offset dimensions in `Out` according to the `collapsed_slice_dims`
1349 and range [`0`, `operand.rank`) \ `collapsed_slice_dims`. So if, e.g.,
1350 `offset.size` is `4`, `operand.rank` is `6` and `collapsed_slice_dims` is {`0`,
1367 - We reshape the slice by collapsing the `collapsed_slice_dims` dimensions.
1427 are listed explicitly as `collapsed_slice_dims`, must have a slice size of
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto449 // if k in collapsed_slice_dims
453 repeated int64 collapsed_slice_dims = 2; field
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1956 self.collapsed_slice_dims = []