Searched refs:stride_int (Results 1 – 1 of 1) sorted by relevance
50 std::vector<int32> stride_int; in PoolingOp() local56 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_int)); in PoolingOp()57 OP_REQUIRES(ctx, stride_int.size() == num_dims(), in PoolingOp()63 stride_.push_back(stride_int[i]); in PoolingOp()