Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h362 static StatusOr<Shape> InferSetDimensionSizeShape(const Shape& operand_shape,
Dshape_inference_test.cc1665 auto inferred_status = ShapeInference::InferSetDimensionSizeShape( in TEST_F()
1676 auto inferred_status = ShapeInference::InferSetDimensionSizeShape( in TEST_F()
Dhlo_verifier.cc1695 ShapeInference::InferSetDimensionSizeShape( in HandleSetDimensionSize()
Dshape_inference.cc2429 /* static */ StatusOr<Shape> ShapeInference::InferSetDimensionSizeShape( in InferSetDimensionSizeShape() function in xla::ShapeInference
Dhlo_parser.cc2953 return ShapeInference::InferSetDimensionSizeShape( in CreateInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3598 ShapeInference::InferSetDimensionSizeShape( in SetDimensionSize()