Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dslice_spmd_expander.cc110 llvm::SmallVector<int64_t, 4> CalculateBitVector(const uint64_t mask_value) { in CalculateBitVector() function
392 CalculateBitVector(slice_op.new_axis_mask()); in ApplyNewAndShrinkMasksToLayout()
394 CalculateBitVector(slice_op.shrink_axis_mask()); in ApplyNewAndShrinkMasksToLayout()
620 CalculateBitVector(strided_slice_op.new_axis_mask()); in ExpandOp()
622 CalculateBitVector(strided_slice_op.shrink_axis_mask()); in ExpandOp()