| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 231 const Tensor& output_backprop = context->input(0); in Compute() local
234 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
236 output_backprop.shape().DebugString())); in Compute()
240 FastBoundsCheck(output_backprop.NumElements(), in Compute()
245 GetBiasValueDims(output_backprop, data_format_, &batch, &height, &width, in Compute()
253 } else if (output_backprop.NumElements() == 0) { in Compute()
266 redux(context->eigen_device<Device>(), three_dims, output_backprop, in Compute()
275 redux(context->eigen_device<Device>(), two_dims, output_backprop, in Compute()
459 const Tensor& output_backprop, int32 batch, in ComputeWithCustomKernel() argument
463 output_backprop.template flat<T>().data(), in ComputeWithCustomKernel()
[all …]
|
| D | bias_op_gpu.cu.cc | 104 const T* __restrict__ output_backprop, in BiasGradNHWC_Naive() argument
109 GpuAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNHWC_Naive()
116 const T* __restrict__ output_backprop, in BiasGradNCHW_Naive() argument
122 GpuAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNCHW_Naive()
128 int32 nthreads, const T* __restrict__ output_backprop, in BiasGradNHWC_SharedAtomics() argument
141 GpuAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()
152 const T* __restrict__ output_backprop, T* __restrict__ bias_backprop, in BiasGradNCHW_SharedAtomics() argument
173 T val = ldg(output_backprop + in BiasGradNCHW_SharedAtomics()
212 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute() argument
236 output_backprop, bias_backprop, bias_size)); in compute()
[all …]
|
| D | bias_op_gpu.h | 40 static void compute(const GPUDevice& device, const T* output_backprop,
|
| D | cudnn_rnn_ops.cc | 855 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward() argument
889 auto output_backprop_data = AsDeviceMemory<T>(output_backprop); in DoBackward()
1671 Tensor output_backprop; in MaybeAutoTune() local
1680 data_type, model_shapes.output_shape, &output_backprop)); in MaybeAutoTune()
1736 input_c, params, output, output_h, output_c, &output_backprop, in MaybeAutoTune()
1886 const Tensor* output_backprop = nullptr; in ComputeImpl() local
1893 &output_backprop, &output_h_backprop, in ComputeImpl()
1920 input_c, params, output, output_h, output_c, output_backprop, in ComputeImpl()
1945 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs() argument
1948 TF_RETURN_IF_ERROR(context->input("output_backprop", output_backprop)); in ExtractBackwardInputs()
[all …]
|
| D | nn_ops_test.cc | 955 Tensor output_backprop(DT_FLOAT, output_shape); in BM_AvgPoolBk() local
956 test::FillIota<float>(&output_backprop, 11.0); in BM_AvgPoolBk()
957 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 | pooling_ops_test.py | 1522 def _testMaxPoolGradDirect(self, input_data, output_backprop, argument
1534 output_backprop, shape=output_sizes)
1556 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1566 output_backprop,
1583 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1593 output_backprop,
1624 output_backprop = [
1651 output_backprop,
1668 output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
1678 output_backprop,
[all …]
|
| 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,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CudnnRNNBackpropV3.pbtxt | 36 name: "output_backprop"
182 name: "output_backprop"
335 name: "output_backprop"
|
| D | CudnnRNNBackpropV2.pbtxt | 32 name: "output_backprop"
|
| D | CudnnRNNBackprop.pbtxt | 32 name: "output_backprop"
|
| /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/ops/compat/ops_history_v2/ |
| D | CudnnRNNBackpropV3.pbtxt | 36 name: "output_backprop"
182 name: "output_backprop"
|
| D | CudnnRNNBackpropV2.pbtxt | 32 name: "output_backprop"
|
| D | CudnnRNNBackprop.pbtxt | 32 name: "output_backprop"
|
| /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/v2/ |
| D | tensorflow.raw_ops.pbtxt | 949 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
953 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
957 …s\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 949 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
953 …, \'input_c\', \'params\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
957 …s\', \'sequence_lengths\', \'output\', \'output_h\', \'output_c\', \'output_backprop\', \'output_h…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 5502 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
5513 …input, input_h, input_c, params, sequence_lengths, output, output_h, output_c, output_backprop, ou…
5613 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
5624 …input, input_h, input_c, params, output, output_h, output_c, output_backprop, output_h_backprop, o…
11278 …tf.Output, output tf.Output, output_h tf.Output, output_c tf.Output, output_backprop tf.Output, ou…
11289 …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 | 9664 name: "output_backprop"
9802 name: "output_backprop"
9948 name: "output_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 16774 name: "output_backprop"
16912 name: "output_backprop"
17058 name: "output_backprop"
|