Searched refs:shape_y (Results 1 – 3 of 3) sorted by relevance
27 self.shape_y = self.inputs[1]['shape']32 if self.shape_x != self.shape_y:
458 def check_two_shapes_need_broadcast(shape_x, shape_y): argument460 if any(j not in (i, 1) for i, j in zip(reversed(shape_x), reversed(shape_y))):462 return shape_y != shape_x
386 ShapePtr shape_y = dyn_cast<Shape>(args_spec_list[0]->GetShapeTrack()); in InferImplBiasAddGrad() local387 MS_EXCEPTION_IF_NULL(shape_y); in InferImplBiasAddGrad()388 ShapeVector y_dims = shape_y->shape(); in InferImplBiasAddGrad()