Searched refs:pad_total (Results 1 – 3 of 3) sorted by relevance
500 const int64 pad_total = padded_in_size - dims.spatial_dims[i].input_size; in MakeXlaBackpropFilterConvOp() local515 ? std::max<int64>(pad_total / 2, 0) in MakeXlaBackpropFilterConvOp()517 padding[i] = {pad_before, pad_total - pad_before}; in MakeXlaBackpropFilterConvOp()
168 int32_t pad_total = kernel_size - 1; in HandlePad() local169 int32_t pad_beg = (pad_total / 2 + 1) / block_size; in HandlePad()170 int32_t pad_end = (pad_total / 2) / block_size; in HandlePad()
4410 const int64_t pad_total = padded_in_size - spatial_dim_i.input_size; in matchAndRewrite() local4425 ? std::max<int64_t>(pad_total / 2, 0) in matchAndRewrite()4428 paddings.push_back(pad_total - pad_before); in matchAndRewrite()