Searched refs:src_weights_shape (Results 1 – 1 of 1) sorted by relevance
479 const auto& src_weights_shape = src_weights_array.shape(); in ConvertDepthwiseConvOperator() local480 CHECK_EQ(src_weights_shape.dimensions_count(), 4); in ConvertDepthwiseConvOperator()482 Shape({src_weights_shape.dims(1), src_weights_shape.dims(2), in ConvertDepthwiseConvOperator()483 src_weights_shape.dims(3) / src_op.depth_multiplier, in ConvertDepthwiseConvOperator()485 CHECK_EQ(src_weights_shape.dims(3) % src_op.depth_multiplier, 0); in ConvertDepthwiseConvOperator()487 src_weights_shape.dims(3)); in ConvertDepthwiseConvOperator()488 CHECK_EQ(src_weights_shape.dims(0), 1); in ConvertDepthwiseConvOperator()