Searched refs:shape_i_op (Results 1 – 2 of 2) sorted by relevance
481 def shape_i_op(i): function483 if key not in shape_i_op.cache:484 shape_i_op.cache[key] = Shape_i(i)485 return shape_i_op.cache[key]486 shape_i_op.cache = {}
24 from theano.compile.ops import shape_i, shape_i_op966 ishape = [shape_i_op(i)(img) for i in range(img.ndim)]967 kshape = [shape_i_op(i)(kerns) for i in range(kerns.ndim)]3323 ishape = [shape_i_op(i)(img) for i in range(img.ndim)]3324 tshape = [shape_i_op(i)(topgrad) for i in range(topgrad.ndim)]3359 tshape = [shape_i_op(i)(topgrad) for i in range(topgrad.ndim)]3360 kshape = [shape_i_op(i)(kerns) for i in range(kerns.ndim)]3428 ishape = [shape_i_op(i)(img) for i in range(img.ndim)]3429 tshape = [shape_i_op(i)(topgrad) for i in range(topgrad.ndim)]3467 tshape = [shape_i_op(i)(topgrad) for i in range(topgrad.ndim)][all …]