Home
last modified time | relevance | path

Searched refs:xor_masks (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc147 int64 dimension_to_sort_bound, absl::Span<const int64> xor_masks, in EmitTiledCompareLoop() argument
228 for (int64 xor_mask : xor_masks) { in EmitTiledCompareLoop()
296 absl::string_view name, absl::Span<const int64> xor_masks, in EmitSortInPlace() argument
329 if (xor_masks.size() > 1) { in EmitSortInPlace()
361 if (xor_masks.size() > 1) { in EmitSortInPlace()
365 keys_index, dimension_to_sort, dimension_to_sort_bound, xor_masks, in EmitSortInPlace()
386 xor_masks[0], tiles_index.GetType(), element_address, write_element, in EmitSortInPlace()
Dsort_util.h39 absl::string_view name, absl::Span<const int64> xor_masks,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2983 auto emit_kernel = [&](absl::Span<const int64> xor_masks) { in EmitSortFromMlir() argument
2986 absl::StrJoin(xor_masks, ", ", [](std::string* out, int64 xor_mask) { in EmitSortFromMlir()
2994 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in EmitSortFromMlir()
3008 dimension_to_sort, values_arrays, IrName(context.name), xor_masks, &b_, in EmitSortFromMlir()
3010 xor_masks.size() > 1 ? num_iterations_in_sort_dim in EmitSortFromMlir()
3017 std::vector<int64> xor_masks; in EmitSortFromMlir() local
3027 if (!xor_masks.empty()) { in EmitSortFromMlir()
3028 TF_RETURN_IF_ERROR(emit_kernel(xor_masks)); in EmitSortFromMlir()
3029 xor_masks.clear(); in EmitSortFromMlir()
3033 xor_masks.push_back(xor_mask); in EmitSortFromMlir()
[all …]