Searched refs:src_weights_shape (Results 1 – 1 of 1) sorted by relevance
475 const auto& src_weights_shape = src_weights_array.shape(); in ConvertDepthwiseConvOperator() local476 CHECK_EQ(src_weights_shape.dimensions_count(), 4); in ConvertDepthwiseConvOperator()478 Shape({src_weights_shape.dims(1), src_weights_shape.dims(2), in ConvertDepthwiseConvOperator()479 src_weights_shape.dims(3) / src_op.depth_multiplier, in ConvertDepthwiseConvOperator()481 CHECK_EQ(src_weights_shape.dims(3) % src_op.depth_multiplier, 0); in ConvertDepthwiseConvOperator()483 src_weights_shape.dims(3)); in ConvertDepthwiseConvOperator()484 CHECK_EQ(src_weights_shape.dims(0), 1); in ConvertDepthwiseConvOperator()