Searched refs:grad_output_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander_test.cc | 96 Shape grad_output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F() local
112 HloInstruction::CreateParameter(4, grad_output_shape, "grad_output")); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops.cc | 364 const ::tensorflow::TensorShape& grad_output_shape = grad_output.shape(); in Compute() local
368 OP_REQUIRES(ctx, grad_output_shape == resampler_output_shape, in Compute()
373 grad_output_shape.DebugString())); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2105 TF_ASSIGN_OR_RETURN(const Shape& grad_output_shape, GetShape(grad_output)); in BatchNormGrad()
2109 batch_var_shape, grad_output_shape, feature_index)); in BatchNormGrad()
|