Searched refs:output_grad_shape (Results 1 – 3 of 3) sorted by relevance
632 TensorShape output_grad_shape = ctx->InputShape("grad_output"); in Compile() local634 ctx, output_grad_shape.dims() >= 2, in Compile()636 output_grad_shape.DebugString())); in Compile()
1418 const Shape& output_grad_shape, int64 feature_index) { in InferBatchNormGradShape() argument1425 ExpectArray(output_grad_shape, "output_grad input of batch norm grad")); in InferBatchNormGradShape()1432 ShapeUtil::ValidateShapeWithOptionalLayout(output_grad_shape)); in InferBatchNormGradShape()1442 if (operand_shape.rank() != output_grad_shape.rank()) { in InferBatchNormGradShape()1447 operand_shape.rank(), output_grad_shape.rank()); in InferBatchNormGradShape()1478 if (!ShapeUtil::ElementIsFloating(output_grad_shape)) { in InferBatchNormGradShape()1482 PrimitiveType_Name(output_grad_shape.element_type())); in InferBatchNormGradShape()1485 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(output_grad_shape, in InferBatchNormGradShape()1491 PrimitiveType_Name(output_grad_shape.element_type()), in InferBatchNormGradShape()1557 ShapeUtil::GetDimension(output_grad_shape, i)) { in InferBatchNormGradShape()[all …]
105 const Shape& output_grad_shape,