Searched defs:dynamic_dimension_inference (Results 1 – 2 of 2) sorted by relevance
159 DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize()354 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSplitInput()568 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeCombineInput()719 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSingleGroup()756 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteReverse()906 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionInputGrad()974 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionForward()1036 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionKernelGrad()1121 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReduceWindowSamePadding()1167 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicSelectAndScatterSamePadding()[all …]
141 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference()145 DynamicDimensionInference* dynamic_dimension_inference() { in dynamic_dimension_inference() function