Searched defs:bias_buffer (Results 1 – 11 of 11) 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
127 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run() local
74 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine() local
691 se::DeviceMemoryBase bias_buffer, in DoMatmul()720 se::DeviceMemoryBase d_buffer, se::DeviceMemoryBase bias_buffer, in ExecuteOnStream()
34 BufferAllocation::Slice bias_buffer) in CublasLtMatmulThunk()
275 se::DeviceMemoryBase bias_buffer; in DoGemmAutotune() local
668 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_F() local778 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local883 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local
93 const BufferT* bias_buffer = model->buffers[bias_tensor->buffer].get(); in DuplicateBiasesWithMultipleUses() local
184 se::DeviceMemoryBase bias_buffer, se::DeviceMemoryBase side_input_buffer, in LogFusedConvForwardAutotuneResults()
471 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local577 const auto bias_buffer = model_.buffers[bias_tensor->buffer].get(); in TEST_P() local