Home
last modified time | relevance | path

Searched refs:in_backprop_flat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc459 auto in_backprop_flat = in_backprop->template flat<T>(); in Compute() local
460 T* input_backprop_data = in_backprop_flat.data(); in Compute()
461 in_backprop_flat.device(context->eigen_device<Device>()) = in Compute()
462 in_backprop_flat.constant(T(0)); in Compute()
Dconv_grad_ops_3d.cc463 auto in_backprop_flat = in_backprop->template flat<T>(); in Compute() local
464 T* input_backprop_data = in_backprop_flat.data(); in Compute()
465 in_backprop_flat.device(context->eigen_device<Device>()) = in Compute()
466 in_backprop_flat.constant(T(0)); in Compute()