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.cc1307 context, num_params_biases_ == rnn_desc->ParamsBiasRegions().size(), in Compute()
1310 rnn_desc->ParamsBiasRegions().size())); in Compute()
1311 for (int i = 0; i < rnn_desc->ParamsBiasRegions().size(); i++) { in Compute()
1312 int64 size_in_bytes = rnn_desc->ParamsBiasRegions()[i].size; in Compute()
1323 input_ptr, rnn_desc->ParamsBiasRegions()[i].offset, size_in_bytes); in Compute()
1397 RestoreParams<T>(biases, rnn_desc->ParamsBiasRegions(), &output_ptr, in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h167 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); } in ParamsBiasRegions() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1224 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1823 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::MIOpenRnnDescriptor