Home
last modified time | relevance | path

Searched refs:InferBatchNormInferenceShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h95 static StatusOr<Shape> InferBatchNormInferenceShape(
Dhlo_verifier.cc1050 ShapeInference::InferBatchNormInferenceShape( in HandleBatchNormInference()
Dshape_inference.cc1315 /* static */ StatusOr<Shape> ShapeInference::InferBatchNormInferenceShape( in InferBatchNormInferenceShape() function in xla::ShapeInference
Dhlo_parser.cc2011 return ShapeInference::InferBatchNormInferenceShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2675 ShapeInference::InferBatchNormInferenceShape( in BatchNormInference()