Searched refs:output_grad_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 632 TensorShape output_grad_shape = ctx->InputShape("grad_output"); in Compile() local
634 ctx, output_grad_shape.dims() >= 2, in Compile()
636 output_grad_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1462 const Shape& output_grad_shape, int64 feature_index) { in InferBatchNormGradShape() argument
1469 ExpectArray(output_grad_shape, "output_grad input of batch norm grad")); in InferBatchNormGradShape()
1476 ShapeUtil::ValidateShapeWithOptionalLayout(output_grad_shape)); in InferBatchNormGradShape()
1486 if (operand_shape.rank() != output_grad_shape.rank()) { in InferBatchNormGradShape()
1491 operand_shape.rank(), output_grad_shape.rank()); in InferBatchNormGradShape()
1522 if (!ShapeUtil::ElementIsFloating(output_grad_shape)) { in InferBatchNormGradShape()
1526 PrimitiveType_Name(output_grad_shape.element_type())); in InferBatchNormGradShape()
1529 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(output_grad_shape, in InferBatchNormGradShape()
1535 PrimitiveType_Name(output_grad_shape.element_type()), in InferBatchNormGradShape()
1601 ShapeUtil::GetDimension(output_grad_shape, i)) { in InferBatchNormGradShape()
[all …]
|
| D | shape_inference.h | 105 const Shape& output_grad_shape,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 862 ShapeHandle output_grad_shape = c->input(2); in Conv2DBackpropInputShape() local
863 TF_RETURN_IF_ERROR(c->WithRank(output_grad_shape, 4, &output_grad_shape)); in Conv2DBackpropInputShape()
871 output_grad_shape, data_format, &batch_size_dim, in Conv2DBackpropInputShape()
|