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

  /external/tensorflow/tensorflow/core/grappler/optimizers/
auto_parallel_test.cc 42 Output apply_gradient = ops::ApplyGradientDescent(
  /external/tensorflow/tensorflow/core/kernels/
training_ops_gpu.cu.cc 29 struct ApplyGradientDescent<GPUDevice, T> {
261 template struct functor::ApplyGradientDescent<GPUDevice, Eigen::half>;
262 template struct functor::ApplyGradientDescent<GPUDevice, float>;
263 template struct functor::ApplyGradientDescent<GPUDevice, double>;
training_ops.h 31 struct ApplyGradientDescent {
training_ops.cc 48 struct ApplyGradientDescent<CPUDevice, T> {
428 functor::ApplyGradientDescent<Device, T>()(
487 Name("ApplyGradientDescent").Device(DEVICE_##D).TypeConstraint<T>("T"), \
505 void ApplyGradientDescent<GPUDevice, T>::operator()( \
509 extern template struct ApplyGradientDescent<GPUDevice, T>;
    [all...]

Completed in 160 milliseconds