HomeSort by relevance Sort by last modified time
    Searched refs:ReluGrad (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
relu_op_gpu.cu.cc 34 template struct functor::ReluGrad<GPUDevice, T>; \
relu_op.cc 41 Name("ReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
82 void ReluGrad<GPUDevice, T>::operator()( \
86 extern template struct ReluGrad<GPUDevice, T>; \
136 Name("ReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
169 Name("ReluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
relu_op_functor.h 41 struct ReluGrad {
42 // Computes ReluGrad backprops.
relu_op.h 87 functor::ReluGrad<Device, T> functor;
  /external/tensorflow/tensorflow/core/ops/
nn_grad.cc 52 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) {
63 {{"dx"}, "ReluGrad", {"dy", "x"}, {{"T", "$T"}}}
68 REGISTER_OP_GRADIENT("Relu", ReluGrad);
  /external/tensorflow/tensorflow/core/profiler/g3doc/
profile_memory.md 34 ReluGrad 9206.45MB (38.48%, 7.65%)
  /external/tensorflow/tensorflow/cc/gradients/
nn_grad.cc 65 auto dx = internal::ReluGrad(scope, grad_inputs[0], op.input(0));
nn_grad_test.cc 111 TEST_F(NNGradTest, ReluGrad) {
  /external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc     [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 518 milliseconds