Searched defs:dynamic_dimension_inference (Results 1 – 2 of 2) sorted by relevance
155 DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize()344 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSplitInput()558 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeCombineInput()709 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSingleGroup()808 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionInputGrad()875 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionForward()936 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConvolutionKernelGrad()1020 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReduceWindowSamePadding()1066 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicSelectAndScatterSamePadding()1146 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicConcat()[all …]
132 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference()136 DynamicDimensionInference* dynamic_dimension_inference() { in dynamic_dimension_inference() function