Searched refs:InferWindowFromDimensions (Results 1 – 6 of 6) sorted by relevance
76 Window window, ShapeInference::InferWindowFromDimensions( in HandleReduce()
337 static StatusOr<Window> InferWindowFromDimensions(
1116 ShapeInference::InferWindowFromDimensions( in TEST_F()1178 ShapeInference::InferWindowFromDimensions( in TEST_F()
2897 ShapeInference::InferWindowFromDimensions( in TEST_P()2948 ShapeInference::InferWindowFromDimensions( in TEST_P()
2348 /* static */ StatusOr<Window> ShapeInference::InferWindowFromDimensions( in InferWindowFromDimensions() function in xla::ShapeInference
1471 ShapeInference::InferWindowFromDimensions( in ConvGeneralDilated()1503 TF_ASSIGN_OR_RETURN(Window window, ShapeInference::InferWindowFromDimensions( in DynamicConvInstruction()2516 ShapeInference::InferWindowFromDimensions( in ReduceWindow()2570 ShapeInference::InferWindowFromDimensions( in ReduceWindowWithGeneralPadding()2614 ShapeInference::InferWindowFromDimensions( in ReduceWindowInternal()3040 ShapeInference::InferWindowFromDimensions( in SelectAndScatter()3087 ShapeInference::InferWindowFromDimensions( in SelectAndScatterInternal()