Searched refs:params_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 761 auto params_data = AsDeviceMemory<T>(params); in DoForward() local
773 input_h_data, *state_desc, input_c_data, params_data, in DoForward()
822 auto params_data = AsDeviceMemory<T>(params); in DoBackward() local
852 params_data, *output_desc, output_data, *state_desc, in DoBackward()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2974 const ParamsT* params_data, in GatherNd() argument
3004 std::memcpy(output_data + i * slice_size, params_data + from_pos, in GatherNd()
|