Home
last modified time | relevance | path

Searched refs:InferConvolveShape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc110 ShapeInference::InferConvolveShape( in TEST_F()
150 ShapeInference::InferConvolveShape( in TEST_F()
295 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
340 ShapeInference::InferConvolveShape(output->shape(), kernel->shape(), in TEST_F()
374 ShapeInference::InferConvolveShape( in TEST_F()
431 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
481 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
535 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
590 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc242 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
298 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
359 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
426 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
Dshape_inference.h110 static StatusOr<Shape> InferConvolveShape(
Dhlo_creation_utils.cc89 ShapeInference::InferConvolveShape( in MakeConvolveHlo()
Dshape_inference_test.cc426 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()
472 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()
518 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()
557 auto inferred_status = ShapeInference::InferConvolveShape( in TEST_F()
Dhlo_verifier.cc157 ShapeInference::InferConvolveShape( in HandleConvolution()
Dshape_inference.cc1571 /* static */ StatusOr<Shape> ShapeInference::InferConvolveShape( in InferConvolveShape() function in xla::ShapeInference
Dalgebraic_simplifier_test.cc3124 ShapeInference::InferConvolveShape(lhs_pad->shape(), filter->shape(), in TEST_P()
3241 ShapeInference::InferConvolveShape(input->shape(), rhs_pad->shape(), in TEST_P()
Dhlo_evaluator_typed_visitor.h1038 ShapeInference::InferConvolveShape( in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc532 ShapeInference::InferConvolveShape( in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1192 Shape shape, ShapeInference::InferConvolveShape( in ConvGeneralDilated()