| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 246 const Tensor& output_backprop = context->input(0); in Compute() local
249 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
251 output_backprop.shape().DebugString())); in Compute()
255 FastBoundsCheck(output_backprop.NumElements(), in Compute()
260 GetBiasValueDims(output_backprop, data_format_, &batch, &height, &width, in Compute()
268 } else if (output_backprop.NumElements() == 0) { in Compute()
284 output_backprop.flat<T>() in Compute()
297 redux(context->eigen_device<Device>(), two_dims, output_backprop, in Compute()
491 const Tensor& output_backprop, int32 batch, in ComputeWithCustomKernel() argument
495 output_backprop.template flat<T>().data(), in ComputeWithCustomKernel()
[all …]
|
| D | bias_op_gpu.cu.cc | 102 __global__ void BiasGradNHWC_Naive(int32 nthreads, const T* output_backprop, in BiasGradNHWC_Naive() argument
106 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNHWC_Naive()
112 __global__ void BiasGradNCHW_Naive(int32 nthreads, const T* output_backprop, in BiasGradNCHW_Naive() argument
118 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNCHW_Naive()
126 const T* output_backprop, in BiasGradNHWC_SharedAtomics() argument
138 CudaAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()
148 __global__ void BiasGradNCHW_SharedAtomics(const T* output_backprop, in BiasGradNCHW_SharedAtomics() argument
171 T val = ldg(output_backprop + in BiasGradNCHW_SharedAtomics()
197 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute() argument
220 d.stream()>>>(total_count, output_backprop, bias_backprop, in compute()
[all …]
|
| D | cudnn_rnn_ops.cc | 796 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward() argument
829 auto output_backprop_data = AsDeviceMemory<T>(output_backprop); in DoBackward()
1485 Tensor output_backprop; in MaybeAutoTune() local
1494 data_type, model_shapes.output_shape, &output_backprop)); in MaybeAutoTune()
1549 input_c, params, output, output_h, output_c, &output_backprop, in MaybeAutoTune()
1687 const Tensor* output_backprop = nullptr; in ComputeImpl() local
1694 &output_backprop, &output_h_backprop, in ComputeImpl()
1721 input_c, params, output, output_h, output_c, output_backprop, in ComputeImpl()
1746 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs() argument
1749 TF_RETURN_IF_ERROR(context->input("output_backprop", output_backprop)); in ExtractBackwardInputs()
[all …]
|
| D | pooling_ops_3d_sycl.h | 255 T* output_backprop = ConvertToActualTypeSycl(T, output_backprop_accessor_);
334 output_backprop[index] = output_value;
426 T* output_backprop = ConvertToActualTypeSycl(T, output_backprop_accessor_);
460 output_backprop[index] = input_backprop[n * p_.in_planes_ * p_.in_rows_ *
655 T* output_backprop = ConvertToActualTypeSycl(T, output_backprop_accessor_);
708 output_backprop[index] = gradient;
|
| D | bias_op_gpu.h | 40 static void compute(const GPUDevice& device, const T* output_backprop,
|
| D | nn_ops_test.cc | 932 Tensor output_backprop(DT_FLOAT, output_shape); in BM_AvgPoolBk() local
933 test::FillIota<float>(&output_backprop, 11.0); in BM_AvgPoolBk()
934 inputs.push_back({nullptr, &output_backprop}); in BM_AvgPoolBk()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 38 output_backprop=grads[0],
63 output_backprop=grad[0],
92 output_backprop=grads[0],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | fractional_max_pool_op_test.py | 380 output_backprop = self._PRNG.randint(100, size=output_data.shape)
382 input_tensor, output_tensor, output_backprop, window_size,
390 output_backprop,
415 output_backprop = self._PRNG.randint(100, size=output_data.shape)
417 input_tensor, output_tensor, output_backprop, window_size,
427 output_backprop,
549 output_backprop = [12.0, 15.0,
578 grad = constant_op.constant(output_backprop, shape=output_size)
|
| D | fractional_avg_pool_op_test.py | 373 output_backprop = (self._PRNG.rand(num_elements) *
376 input_tensor.get_shape(), output_backprop, window_size,
383 output_backprop,
412 output_backprop = (self._PRNG.rand(num_elements) *
415 input_tensor.get_shape(), output_backprop, window_size,
424 output_backprop,
|
| D | pooling_ops_test.py | 1274 def _testMaxPoolGradDirect(self, input_data, output_backprop, argument
1286 output_backprop, shape=output_sizes)
1308 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1318 output_backprop,
1335 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1345 output_backprop,
1376 output_backprop = [
1403 output_backprop,
1418 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1427 output_backprop,
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 85 NodeDef* output_backprop = graph.add_node(); in TEST() local
86 name_to_node["output_backprop"] = output_backprop; in TEST()
88 output_backprop); in TEST()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 30 output_backprop: A 3-D tensor with the same shape as output in the forward pass.
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 33 output_backprop: A 3-D tensor with the same shape as output in the forward pass.
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 37 output_backprop: A 3-D tensor with the same shape as output in the forward pass.
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 737 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
741 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
745 …s\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 737 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
741 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
745 …s\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4798 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
4809 …input, input_h, input_c, params, sequence_lengths, output, output_h, output_c, output_backprop, ou…
4912 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
4923 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
21878 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
21889 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7370 name: "output_backprop"
7508 name: "output_backprop"
7654 name: "output_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 16832 name: "output_backprop"
16970 name: "output_backprop"
17116 name: "output_backprop"
17262 name: "output_backprop"
|
| D | ops_history.v2.pbtxt | 16774 name: "output_backprop"
16912 name: "output_backprop"
17058 name: "output_backprop"
|