Searched refs:bias_buf (Results 1 – 6 of 6) sorted by relevance
44 bias_buf = in SetUp()46 (NN_MAX_HIDDEN_LAYERS + 1) * sizeof(*bias_buf)); in SetUp()48 ASSERT_NE(bias_buf, nullptr); in SetUp()51 bias[i] = &bias_buf[i * NN_MAX_NODES_PER_LAYER]; in SetUp()57 aom_free(bias_buf); in TearDown()71 float *weights_buf = nullptr, *bias_buf = nullptr; member in __anonf412f50d0111::NnPredictTest
21 static const AccumScalar bias_buf[kBufferSize] = {0}; in Run() local24 mul_params->set_bias(bias_buf); in Run()33 static const AccumScalar bias_buf[kBufferSize] = {0}; in Run() local34 mul_params->set_bias(bias_buf); in Run()
424 const float* bias_buf = static_cast<float*>( in GetBiasHandle() local441 ((bias_buf[j] * out_scale) + static_cast<float>(x * qa_amin)); in GetBiasHandle()460 ((bias_buf[j] * out_scale) + static_cast<float>(x * qa_amin)); in GetBiasHandle()476 void* bias_buf = static_cast<void*>( in GetBiasHandle() local479 this->cpu_engine_, bias_buf); in GetBiasHandle()
1552 void* bias_buf = static_cast<void*>( in GetBiasHandle() local1555 input_bias_ = new memory(bias_md, this->cpu_engine_, bias_buf); in GetBiasHandle()1557 input_bias_->set_data_handle(bias_buf); in GetBiasHandle()
85 se::DeviceMemoryBase bias_buf; member
133 DeviceMemory<BiasType>(params.fusion->bias_buf), in RunGpuConvForwardActivation()486 fusion.bias_buf = operand_buffers[2]; in GetGpuConvParams()