Searched defs:n_input_plane (Results 1 – 4 of 4) sorted by relevance
141 void Col2ImCheckNInputPlane(const PrimitivePtr &primitive, int64_t n_input_plane, in Col2ImCheckNInputPlane()319 auto n_input_plane = input_shape[input_shape.size() - kIndex2]; in REGISTER_SIMPLE_INFER() local354 auto n_input_plane = input_shape[input_shape.size() - kIndex2]; in CheckValidation() local424 auto n_input_plane = input_shape[input_rank - kIndex2]; in InferShape() local
94 const int64_t n_input_plane = x_shape_[kIndex1]; in LaunchKernel() local
74 const int64_t n_input_plane = x_shape[kInputIndex2]; in Col2ImShapeCheck() local
225 const int64_t n_input_plane = input_shape[kIndex1]; in Col2imCompute() local