| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 253 HloInstruction* accumulator, absl::Span<const int64> offset_dims, in PermuteBatchAndOffsetDims() argument
259 int64 offset_idx_counter = output_rank - offset_dims.size(); in PermuteBatchAndOffsetDims()
261 bool is_offset_dim = absl::c_binary_search(offset_dims, i); in PermuteBatchAndOffsetDims()
368 AsInt64Slice(dim_numbers.offset_dims()), in ExpandInstruction()
|
| D | shape_inference.cc | 2869 if (!absl::c_is_sorted(dim_numbers.offset_dims())) { in ValidateGatherDimensionNumbers()
2872 StrJoin(dim_numbers.offset_dims(), ", ")); in ValidateGatherDimensionNumbers()
2875 if (absl::c_adjacent_find(dim_numbers.offset_dims()) != in ValidateGatherDimensionNumbers()
2876 dim_numbers.offset_dims().end()) { in ValidateGatherDimensionNumbers()
2879 StrJoin(dim_numbers.offset_dims(), ", ")); in ValidateGatherDimensionNumbers()
2887 int64 offset_dim = dim_numbers.offset_dims(i); in ValidateGatherDimensionNumbers()
3013 slice_sizes.size(), StrJoin(gather_dim_numbers.offset_dims(), ","), in InferGatherShape()
3047 absl::c_binary_search(gather_dim_numbers.offset_dims(), i); in InferGatherShape()
|
| D | hlo_evaluator.cc | 789 !absl::c_binary_search(dim_numbers.offset_dims(), i); in IterationSpaceForOutputBatchIndices()
807 absl::c_binary_search(dim_numbers.offset_dims(), i); in IterationSpaceForOutputOffsetIndices()
836 !absl::c_binary_search(dim_numbers_.offset_dims(), i)); in OutputBatchIndexToInputIndex()
963 if (absl::c_binary_search(dim_numbers.offset_dims(), i)) { in OutputOffsetIndexToInputIndex()
|
| D | hlo_instructions.cc | 2323 string offset_dims = in GatherDimensionNumbersToString() local
2325 StrJoin(gather_dimension_numbers_->offset_dims(), ","), "}"); in GatherDimensionNumbersToString()
2336 {offset_dims, collapsed_slice_dims, start_index_map, index_vector_dim}, in GatherDimensionNumbersToString()
2341 absl::Span<const int64> offset_dims, in MakeGatherDimNumbers() argument
2345 for (int64 output_window_dim : offset_dims) { in MakeGatherDimNumbers()
|
| D | hlo_parser_test.cc | 853 …{4,3,2,1,0} %input_tensor, s64[10,9,8,7,5]{4,3,2,1,0} %start_indices), offset_dims={4,5,6,7,8}, co… in CreateTestCases()
1333 …0,29,28,27,26]{8,7,6,5,4,3,2,1,0} gather(input_tensor, start_indices), offset_dims={4,5,6,7,8}, co… in CreateTestCases()
|
| D | indexed_array_analysis.cc | 293 if (!absl::c_binary_search(dim_numbers.offset_dims(), i)) { in ComputeArrayForGather()
|
| D | hlo_parser.cc | 1615 optional<std::vector<int64>> offset_dims; in ParseInstructionRhs() local
1617 &offset_dims}; in ParseInstructionRhs()
1638 /*offset_dims=*/*offset_dims, in ParseInstructionRhs()
|
| D | elemental_ir_emitter.cc | 1888 int64 output_window_dim = dim_numbers.offset_dims(operand_index_dim++); in EmitElementalGather()
1898 if (!absl::c_binary_search(dim_numbers.offset_dims(), i)) { in EmitElementalGather()
|
| D | hlo_instructions.h | 1380 absl::Span<const int64> offset_dims,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 451 // else Out[offset_dims[i++]]
452 repeated int64 offset_dims = 1; field
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1269 <b> `gather(operand, start_indices, offset_dims, collapsed_slice_dims, slice_sizes, start_index_map…
1283 | `offset_dims` | `ArraySlice<int64>` | The set of dimensions in the |
1298 For convenience, we label dimensions in the output array not in `offset_dims`
1317 2. If `i` is present in `offset_dims` (i.e. equal to `offset_dims`[`k`] for
1342 `Out`[`offset_dims`[`k`]] if `k` < `offset_dims.size`
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1220 dnums.offset_dims.append(1)
1221 dnums.offset_dims.append(2)
|
| D | xla_client.py | 1955 self.offset_dims = []
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3398 std::vector<int64> offset_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F() local
3407 .RandomInput(type, offset_dims) in TEST_F()
|