Searched refs:weightDepth (Results 1 – 1 of 1) sorted by relevance
1093 int32_t weightDepth = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1110 weightDepth = weightShape.getDimSize(3); in inferReturnTypeComponents()1145 !ShapedType::isDynamic(weightDepth)) { in inferReturnTypeComponents()1147 int32_t filterSize = (weightDepth - 1) * dilation[2] + 1; in inferReturnTypeComponents()