Searched defs:grad_y (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 64 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 782 auto grad_y = itensor_from_tensor(grad_output, /*from_const_data_ptr*/true); in mkldnn_convolution_backward_input() local
822 const ideep::tensor grad_y = itensor_from_tensor(grad_output, /*from_const_data_ptr*/true); in mkldnn_convolution_backward_weights() local
932 auto grad_y = itensor_from_tensor(grad_output, /*from_const_data_ptr*/true); in mkldnn_convolution_transpose_backward_input() local
973 auto grad_y = itensor_from_tensor(grad_output, /*from_const_data_ptr*/true); in mkldnn_convolution_transpose_backward_weights() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | linalg_grad.cc | 436 Output grad_y = in EinsumGrad() local
|
| D | nn_grad.cc | 430 Output grad_y = grad_inputs[0]; in BaseFusedBatchNormGrad() local
494 std::vector<Output>* grad_outputs) { in FusedBatchNormV3Grad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 928 def _BatchNormGrad(grad_y, argument
|
| /external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/ |
| D | esextcTextureCubeMapArraySampling.hpp | 347 bufferDefinition grad_y; member in glcts::TextureCubeMapArraySamplingTest::bufferCollection
|
| D | esextcTextureCubeMapArraySampling.cpp | 1074 void prepareGradYForFace(glw::GLfloat *grad_y, glw::GLuint face, glw::GLfloat *texture_coordinates,… in prepareGradYForFace()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test.py | 1553 def _testGradient(self, func, x, expected_grad, grad_y=None): argument
|
| /external/pytorch/test/cpp/api/ |
| D | autograd.cpp | 449 auto grad_y = torch::autograd::grad({out}, {y}, {go})[0]; in TEST() local
|
| /external/pytorch/test/functorch/ |
| D | test_eager_transforms.py | 1311 def h(x, grad_y): argument
|
| /external/pytorch/test/ |
| D | test_autograd.py | 2016 def backward(ctx, grad_x, grad_y): argument
|