Searched defs:params_data (Results 1 – 6 of 6) sorted by relevance
217 tpu::kMaxAuxiliaryParameterCount + 1> params_data; in Compute() local345 tpu::kMaxAuxiliaryParameterCount + 1> params_data; in Compute() local
290 def _set_param(self, params_data): argument
604 const ParamsT* params_data, in GatherNd()632 const TfLiteTensor* params_data, in GatherNdString()
157 void LBFGS::_set_param(const std::vector<Tensor>& params_data) { in _set_param()
823 auto params_data = AsDeviceMemory<T>(params); in DoForward() local905 auto params_data = AsDeviceMemory<T>(params); in DoBackward() local
524 int google_chromeec_flash_write_block(const uint8_t *params_data, in google_chromeec_flash_write_block()