Searched refs:grad_data (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops_gpu.cu.cc | 139 atomicAdd(grad_data + (batch_id * data_batch_stride + \
146 const T* __restrict__ grad_output, T* __restrict__ grad_data, in ResamplerGrad2DKernel() argument
245 const T* __restrict__ grad_output, T* __restrict__ grad_data, in operator ()()
264 grad_data_size, grad_data)); in operator ()()
271 warp, grad_output, grad_data, grad_warp, in operator ()()
|
| D | resampler_ops.cc | 208 const T* __restrict__ grad_output, T* __restrict__ grad_data, in operator ()()
219 memset(grad_data, 0, sizeof(T) * grad_data_size); in operator ()()
248 grad_data[batch_id * data_batch_stride + in operator ()()
375 ::tensorflow::Tensor* grad_data = nullptr; in Compute()
377 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, data.shape(), &grad_data)); in Compute()
384 grad_data->flat<T>().data(), grad_warp->flat<T>().data(), batch_size, in Compute()
|
| D | resampler_ops.h | 45 const T* __restrict__ grad_output, T* __restrict__ grad_data,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 167 XlaOp ScatterToGradData(XlaOpKernelContext* ctx, XlaOp grad_data, XlaOp indices, in ScatterToGradData() argument
187 return xla::Scatter(grad_data, indices, updates, in ScatterToGradData()
301 auto grad_data = xla::ConstantLiteral( in CalculateGradData() local
309 xla::Pad(grad_data, xla::Zero(ctx->builder(), warp_type), in CalculateGradData()
658 auto grad_data = CalculateGradData( in Compile() local
674 ctx->SetOutput(0, grad_data); in Compile()
|
| /external/tensorflow/tensorflow/contrib/resampler/xla/ |
| D | resampler_ops_xla_test.py | 49 grad_data, grad_warp = gen_resampler_ops.resampler_grad(
52 grad_data_tf, grad_warp_tf = sess.run([grad_data, grad_warp], {
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cholesky_op_test.py | 258 grad_data = np.random.randn(*data.shape).astype(np.float32)
263 composite_grad = gradients_impl.gradients(chol, x, grad_data)[0]
264 specialized_grad = SpecializedGrad(chol, grad_data)
|