Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc1349 context, num_params_biases_ == rnn_desc->ParamsBiasRegions().size(), in Compute()
1352 rnn_desc->ParamsBiasRegions().size())); in Compute()
1353 for (int i = 0; i < rnn_desc->ParamsBiasRegions().size(); i++) { in Compute()
1354 int64_t size_in_bytes = rnn_desc->ParamsBiasRegions()[i].size; in Compute()
1365 input_ptr, rnn_desc->ParamsBiasRegions()[i].offset, size_in_bytes); in Compute()
1439 RestoreParams<T>(biases, rnn_desc->ParamsBiasRegions(), &output_ptr, in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h175 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); } in ParamsBiasRegions() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1877 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::MIOpenRnnDescriptor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1402 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions() function in stream_executor::gpu::CudnnRnnDescriptor