Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc842 SmallVector<int32_t, 2> a0_pad_const(2 * (input_rank)); in convertSpaceToBatchNDOp() local
854 a0_pad_const[0] = 0; in convertSpaceToBatchNDOp()
855 a0_pad_const[1] = 0; in convertSpaceToBatchNDOp()
862 a0_pad_const[idx + 2] = i.getInt(); in convertSpaceToBatchNDOp()
869 int32_t lo_pad = a0_pad_const[2 * (i + 1) + 0]; in convertSpaceToBatchNDOp()
870 int32_t hi_pad = a0_pad_const[2 * (i + 1) + 1]; in convertSpaceToBatchNDOp()
877 a0_pad_const[2 * (i + block_rank + 1) + 0] = 0; in convertSpaceToBatchNDOp()
878 a0_pad_const[2 * (i + block_rank + 1) + 1] = 0; in convertSpaceToBatchNDOp()
889 llvm::makeArrayRef<int32_t>(a0_pad_const))); in convertSpaceToBatchNDOp()