Home
last modified time | relevance | path

Searched refs:ParamsBiasRegions (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1165 OP_REQUIRES(context, num_params_ == rnn_desc->ParamsBiasRegions().size(), in Compute()
1168 rnn_desc->ParamsBiasRegions().size())); in Compute()
1169 for (int i = 0; i < rnn_desc->ParamsBiasRegions().size(); i++) { in Compute()
1170 int64 size_in_bytes = rnn_desc->ParamsBiasRegions()[i].size; in Compute()
1181 input_ptr, rnn_desc->ParamsBiasRegions()[i].offset, size_in_bytes); in Compute()
1232 RestoreParams<T>(biases, rnn_desc->ParamsBiasRegions(), &output_ptr, in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h175 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); }
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1129 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1742 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::MIOpenRnnDescriptor