Searched refs:InferConvolveShape (Results 1 – 15 of 15) sorted by relevance
110 ShapeInference::InferConvolveShape( in TEST_F()151 ShapeInference::InferConvolveShape( in TEST_F()298 ShapeInference::InferConvolveShape( in TEST_F()343 ShapeInference::InferConvolveShape( in TEST_F()379 ShapeInference::InferConvolveShape( in TEST_F()437 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()488 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()543 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()599 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
243 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()300 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()362 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()430 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
32 return ShapeInference::InferConvolveShape( in MaybeInferShape()
438 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()484 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()530 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()569 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()606 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()664 ShapeInference::InferConvolveShape( in TEST_F()676 ShapeInference::InferConvolveShape( in TEST_F()688 ShapeInference::InferConvolveShape( in TEST_F()700 ShapeInference::InferConvolveShape( in TEST_F()712 ShapeInference::InferConvolveShape( in TEST_F()[all …]
142 ShapeInference::InferConvolveShape( in CreateShardedConvForDotGeneralConvolution()
111 static StatusOr<Shape> InferConvolveShape(
103 ShapeInference::InferConvolveShape( in MakeConvolveHlo()
153 ShapeInference::InferConvolveShape( in HandleConvolution()
3884 ShapeInference::InferConvolveShape( in TEST_P()4003 ShapeInference::InferConvolveShape( in TEST_P()4151 Shape out_shape = ShapeInference::InferConvolveShape( in TEST_F()
1615 /* static */ StatusOr<Shape> ShapeInference::InferConvolveShape( in InferConvolveShape() function in xla::ShapeInference
1291 ShapeInference::InferConvolveShape(
1644 return ShapeInference::InferConvolveShape( in ParseInstructionRhs()
466 ShapeInference::InferConvolveShape( in ConvArray4DGeneralDimensionsDilated()
951 ShapeInference::InferConvolveShape( in CreateShardedConvConvolution()
1476 ShapeInference::InferConvolveShape( in ConvGeneralDilated()1508 ShapeInference::InferConvolveShape( in DynamicConvInstruction()