Home
last modified time | relevance | path

Searched refs:InferWindowFromDimensions (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtree_reduction_rewriter.cc90 Window window, ShapeInference::InferWindowFromDimensions( in HandleReduce()
Dshape_inference.h368 static StatusOr<Window> InferWindowFromDimensions(
Dshape_inference_test.cc1151 ShapeInference::InferWindowFromDimensions( in TEST_F()
1213 ShapeInference::InferWindowFromDimensions( in TEST_F()
Dhlo_evaluator_test.cc2935 ShapeInference::InferWindowFromDimensions( in TEST_P()
2986 ShapeInference::InferWindowFromDimensions( in TEST_P()
Dshape_inference.cc2455 /* static */ StatusOr<Window> ShapeInference::InferWindowFromDimensions( in InferWindowFromDimensions() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc1310 llvm::Optional<xla::Window> InferWindowFromDimensions( in InferWindowFromDimensions() function
1464 auto window = InferWindowFromDimensions( in InferShapeForXlaReduceWindowOp()
1516 auto window = InferWindowFromDimensions( in InferShapeForXlaSelectAndScatterOp()
1572 InferWindowFromDimensions(window_spatial_dims, window_strides, paddings, in InferXlaConvOutputShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1549 ShapeInference::InferWindowFromDimensions( in ConvGeneralDilated()
1583 TF_ASSIGN_OR_RETURN(Window window, ShapeInference::InferWindowFromDimensions( in DynamicConvInstruction()
2670 ShapeInference::InferWindowFromDimensions( in ReduceWindow()
2724 ShapeInference::InferWindowFromDimensions( in ReduceWindowWithGeneralPadding()
2768 ShapeInference::InferWindowFromDimensions( in ReduceWindowInternal()
3270 ShapeInference::InferWindowFromDimensions( in SelectAndScatter()
3317 ShapeInference::InferWindowFromDimensions( in SelectAndScatterInternal()