Searched refs:start_index_map (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 132 int64 index_vector_dim_index = FindIndex(dim_numbers.start_index_map(), i); in ExpandIndexVectorIntoOperandSpace()
|
| D | indexed_array_analysis.cc | 267 dim_numbers.collapsed_slice_dims(0) != dim_numbers.start_index_map(0)) { in ComputeArrayForGather()
290 int64 source_dim = dim_numbers.start_index_map(0); in ComputeArrayForGather()
|
| D | hlo_instructions.cc | 2329 string start_index_map = in GatherDimensionNumbersToString() local
2331 StrJoin(gather_dimension_numbers_->start_index_map(), ","), "}"); in GatherDimensionNumbersToString()
2336 {offset_dims, collapsed_slice_dims, start_index_map, index_vector_dim}, in GatherDimensionNumbersToString()
2343 absl::Span<const int64> start_index_map, int64 index_vector_dim) { in MakeGatherDimNumbers() argument
2351 for (int64 gather_dim_to_input_dim : start_index_map) { in MakeGatherDimNumbers()
|
| D | shape_inference.cc | 2908 int64 operand_dim_for_start_index_i = dim_numbers.start_index_map(i); in ValidateGatherDimensionNumbers()
2918 dim_numbers.start_index_map().begin(), in ValidateGatherDimensionNumbers()
2919 dim_numbers.start_index_map().end()); in ValidateGatherDimensionNumbers()
2928 StrJoin(dim_numbers.start_index_map(), ", ")); in ValidateGatherDimensionNumbers()
|
| D | hlo_evaluator.cc | 841 std::distance(dim_numbers_.start_index_map().begin(), in OutputBatchIndexToInputIndex()
842 absl::c_find(dim_numbers_.start_index_map(), i)); in OutputBatchIndexToInputIndex()
|
| D | hlo_parser_test.cc | 853 …} %start_indices), offset_dims={4,5,6,7,8}, collapsed_slice_dims={}, start_index_map={0,1,2,3,4}, … in CreateTestCases()
1333 …r, start_indices), offset_dims={4,5,6,7,8}, collapsed_slice_dims={}, start_index_map={0,1,2,3,4}, … in CreateTestCases()
|
| D | hlo_parser.cc | 1621 optional<std::vector<int64>> start_index_map; in ParseInstructionRhs() local
1623 &start_index_map}; in ParseInstructionRhs()
1640 /*start_index_map=*/*start_index_map, in ParseInstructionRhs()
|
| D | hlo_instructions.h | 1382 absl::Span<const int64> start_index_map, int64 index_vector_dim);
|
| D | elemental_ir_emitter.cc | 1915 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 473 ? use->gather_dimension_numbers().start_index_map() in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 455 // This is interpreted as a map from i to start_index_map[i]. It
458 repeated int64 start_index_map = 3; 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…
1293 | `start_index_map` | `ArraySlice<int64>` | A map that describes how to |
1333 scattering `S` using `start_index_map`. More precisely:
1334 1. `S`<sub>`in`</sub>[`start_index_map`[`k`]] = `S`[`k`] if `k` <
1335 `start_index_map.size`.
1361 - We use `start_index_map` to map the starting index (which may have size less
1446 these are expanded in accordance to "Gather Index Mapping" (`start_index_map` in
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1222 dnums.start_index_map.append(0)
1223 dnums.start_index_map.append(1)
|
| D | xla_client.py | 1957 self.start_index_map = []
|