| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 41 reserve_space=op.outputs[3],
66 reserve_space=op.outputs[3],
95 reserve_space=op.outputs[3],
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 35 reserve_space: The same reserve_space produced in for forward operation.
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 38 reserve_space: The same reserve_space produced in the forward operation.
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 44 reserve_space: The same reserve_space produced in the forward operation.
|
| D | api_def_CudnnRNN.pbtxt | 33 reserve_space: An opaque tensor that can be used in backprop calculation. It
|
| D | api_def_CudnnRNNV2.pbtxt | 34 reserve_space: An opaque tensor that can be used in backprop calculation. It
|
| D | api_def_CudnnRNNV3.pbtxt | 41 reserve_space: An opaque tensor that can be used in backprop calculation. It
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 797 const Tensor* output_c_backprop, const Tensor* reserve_space, in DoBackward() argument
843 CastDeviceMemory<uint8, T>(const_cast<Tensor*>(reserve_space)); in DoBackward()
1546 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutoTune() local
1550 &output_h_backprop, &output_c_backprop, &reserve_space, in MaybeAutoTune()
1690 const Tensor* reserve_space = nullptr; in ComputeImpl() local
1695 &output_c_backprop, &reserve_space)); in ComputeImpl()
1722 output_h_backprop, output_c_backprop, reserve_space, input_backprop, in ComputeImpl()
1747 const Tensor** output_c_backprop, const Tensor** reserve_space) { in ExtractBackwardInputs() argument
1757 TF_RETURN_IF_ERROR(context->input("reserve_space", reserve_space)); in ExtractBackwardInputs()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1539 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl()
1548 SE_ASSIGN_OR_RETURN(reserve_space, in DoRnnForwardImpl()
1620 /*reserveSpace=*/reserve_space.opaque(), in DoRnnForwardImpl()
1621 /*reserveSpaceSizeInBytes=*/reserve_space.size())); in DoRnnForwardImpl()
1640 /*reserveSpace=*/reserve_space.opaque(), in DoRnnForwardImpl()
1641 /*reserveSpaceSizeInBytes=*/reserve_space.size())); in DoRnnForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2049 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local
2065 (reserve_space = allocated.ValueOrDie()) == nullptr) { in DoRnnForwardImpl()
2069 stream->ThenMemZero(&reserve_space, reserve_space_size_in_bytes); in DoRnnForwardImpl()
2104 reserve_space.opaque() /*reserveSpace*/, in DoRnnForwardImpl()
2105 reserve_space.size() /*reserveSpaceSizeInBytes*/); in DoRnnForwardImpl()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4798 … tf.Output, output_h_backprop tf.Output, output_c_backprop tf.Output, reserve_space tf.Output, hos…
4809 …t_h, output_c, output_backprop, output_h_backprop, output_c_backprop, reserve_space, host_reserved,
4912 … tf.Output, output_h_backprop tf.Output, output_c_backprop tf.Output, reserve_space tf.Output, hos…
4923 …t_h, output_c, output_backprop, output_h_backprop, output_c_backprop, reserve_space, host_reserved,
5025 …RNNV2Attr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserve_space tf.Output, hos…
17225 …RNNV3Attr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserve_space tf.Output, hos…
21878 … tf.Output, output_h_backprop tf.Output, output_c_backprop tf.Output, reserve_space tf.Output, opt…
21889 …, output, output_h, output_c, output_backprop, output_h_backprop, output_c_backprop, reserve_space,
25116 …CudnnRNNAttr) (output tf.Output, output_h tf.Output, output_c tf.Output, reserve_space tf.Output) {
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 737 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'rnn_mode\…
741 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'host_rese…
745 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'host_rese…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 737 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'rnn_mode\…
741 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'host_rese…
745 … \'output_backprop\', \'output_h_backprop\', \'output_c_backprop\', \'reserve_space\', \'host_rese…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7253 name: "reserve_space"
7382 name: "reserve_space"
7520 name: "reserve_space"
7666 name: "reserve_space"
8124 name: "reserve_space"
8249 name: "reserve_space"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 16715 name: "reserve_space"
16844 name: "reserve_space"
16982 name: "reserve_space"
17128 name: "reserve_space"
17274 name: "reserve_space"
17732 name: "reserve_space"
17857 name: "reserve_space"
17982 name: "reserve_space"
|
| D | ops_history.v2.pbtxt | 16657 name: "reserve_space"
16786 name: "reserve_space"
16924 name: "reserve_space"
17070 name: "reserve_space"
17521 name: "reserve_space"
17646 name: "reserve_space"
|