Searched refs:next_grad_size (Results 1 – 1 of 1) sorted by relevance
688 std::vector<int64> next_grad_size = {(in_size[0] - 1) * 2 + 1, in Compile() local691 b, grad, xla::F32, num_spatial_dims, in_size, next_grad_size, in Compile()694 in_size = next_grad_size; in Compile()