Searched defs:ReluGrad (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | relu_op_functor.h | 42 struct ReluGrad { struct
49 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
|
| H A D | relu_op_gpu.cu.cc | 158 struct ReluGrad<Device, Eigen::half> { struct
165 void operator()(const Device& d, in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | nn_grad.cc | 75 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) { in ReluGrad() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/ |
| H A D | nn_ops.cc | 66 Status ReluGrad(AbstractContext* ctx, AbstractTensorHandle* const gradients, in ReluGrad() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | nn_grad_test.cc | 153 TEST_F(NNGradTest, ReluGrad) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 3827 TEST_F(OpTest, ReluGrad) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | wrappers.go | 36455 func ReluGrad(scope *Scope, gradients tf.Output, features tf.Output) (backprops tf.Output) { func
|