Searched refs:sliced_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 111 XlaOp sliced_input = in TopKWithPartitions() local
116 sliced_input = ConcatInDim(builder, {values, sliced_input}, last_dim); in TopKWithPartitions()
121 Sort({sliced_input, sliced_indices}, in TopKWithPartitions()
147 XlaOp sliced_input = Slice(input, start_indices, limit_indices, strides); in TopKWithPartitions() local
152 Sort({sliced_input, sliced_indices}, in TopKWithPartitions()
|
| D | slicing.cc | 32 XlaOp sliced_input = DynamicSlice(input, base_indices, window_sizes); in DynamicStridedSlice() local
35 sliced_input = Slice(sliced_input, std::vector<int64>(window_sizes.size()), in DynamicStridedSlice()
38 return sliced_input; in DynamicStridedSlice()
|