Searched defs:select_val (Results 1 – 2 of 2) sorted by relevance
1651 HloInstruction* select_val, int64 new_batch_dim, int64 new_space_dim, in SelectValidPortion()1895 auto select_val = computation_->AddInstruction(HloInstruction::CreateConstant( in PropagateOnConv() local2372 auto select_val = computation_->AddInstruction(HloInstruction::CreateConstant( in PropagateOnBackpropFilterConv() local2800 auto select_val = computation_->AddInstruction(HloInstruction::CreateConstant( in PerformSpaceToBatchOnConvolution() local
62 uint16_t select_val; member