Home
last modified time | relevance | path

Searched refs:InferReduceShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc65 auto inferred_status = ShapeInference::InferReduceShape( in ExpectInferredReduceShape()
718 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
731 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
743 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
755 auto inferred_status = ShapeInference::InferReduceShape({}, {0, 1}, to_apply); in TEST_F()
766 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
779 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
792 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
803 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()
815 ShapeInference::InferReduceShape({&arg_shape, &f32_}, in TEST_F()
[all …]
Dshape_inference.h151 static StatusOr<Shape> InferReduceShape(
Dhlo_verifier.cc453 CheckShape(reduce, ShapeInference::InferReduceShape( in HandleReduce()
Dshape_inference.cc2042 /* static */ StatusOr<Shape> ShapeInference::InferReduceShape( in InferReduceShape() function in xla::ShapeInference
Dhlo_evaluator_typed_visitor.h1696 ShapeInference::InferReduceShape( in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1969 ShapeInference::InferReduceShape( in Reduce()