Searched defs:bias_buffer (Results 1 – 6 of 6) sorted by relevance
59 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine() local131 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local
71 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in ProcessLinearOperator() local
125 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run() local
74 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine() local
173 se::DeviceMemoryBase bias_buffer, se::DeviceMemoryBase side_input_buffer, in LogFusedConvForwardAutotuneResults()
653 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_F() local762 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local