| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| bfloat16_test.cc | 134 auto grad_output = builder.ConstantR4FromArray4D<bfloat16>( local
140 builder.BatchNormGrad(operand, scale, mean, var, grad_output,
|
| batch_normalization_test.cc | 346 auto grad_output = builder.ConstantR4FromArray4D<float>( local
349 builder.BatchNormGrad(operand, scale, mean, var, grad_output,
804 builder.Parameter(4, grad_output_literal->shape(), "grad_output");
[all...] |
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| resampler_ops.cc | 208 const T* __restrict__ grad_output, T* __restrict__ grad_data,
282 grad_output[batch_id * output_batch_stride +
338 const ::tensorflow::Tensor& grad_output = ctx->input(2); variable
364 const ::tensorflow::TensorShape& grad_output_shape = grad_output.shape();
370 "grad_output shape is not consistent with data and warp "
383 warp.flat<T>().data(), grad_output.flat<T>().data(),
|
| /external/tensorflow/tensorflow/c/ |
| while_loop_test.cc | 433 TF_Output grad_output; local
435 nullptr, s_, &grad_output);
439 Run({grad_output}, {0});
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander.cc | 434 HloInstruction* grad_output = batch_norm->mutable_operand(4); local
498 grad_output, activation_minus_mean));
511 feature_shape, grad_output, zero, dimensions_without_feature,
564 grad_output, elements_per_feature));
|
| user_computation.cc | 612 TF_ASSIGN_OR_RETURN(const OperationRequest* grad_output,
613 LookUpRequest(batch_norm_grad_request.grad_output()));
624 variance->output_shape(), grad_output->output_shape(),
3149 HloInstruction* grad_output = local
[all...] |
