Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h101 static StatusOr<Shape> InferBatchNormGradShape(const Shape& operand_shape,
Dhlo_verifier.cc1060 return CheckShape(batch_norm_grad, ShapeInference::InferBatchNormGradShape( in HandleBatchNormGrad()
Dshape_inference.cc1459 /* static */ StatusOr<Shape> ShapeInference::InferBatchNormGradShape( in InferBatchNormGradShape() function in xla::ShapeInference
Dhlo_parser.cc2038 return ShapeInference::InferBatchNormGradShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2701 Shape shape, ShapeInference::InferBatchNormGradShape( in BatchNormGrad()