Searched refs:instruction_count_after (Results 1 – 1 of 1) sorted by relevance
302 int64 instruction_count_after = computation_->instruction_count(); in HandleBatchNormTraining() local303 CHECK_EQ(instruction_count_after, in HandleBatchNormTraining()407 int64 instruction_count_after = computation_->instruction_count(); in HandleBatchNormInference() local408 CHECK_EQ(instruction_count_after, in HandleBatchNormInference()585 int64 instruction_count_after = computation_->instruction_count(); in HandleBatchNormGrad() local586 CHECK_EQ(instruction_count_after, in HandleBatchNormGrad()