Searched refs:vector_idx (Results 1 – 2 of 2) sorted by relevance
64 const int vector_idx = agent.key().id(); in Lookup() local65 return Optional<LookupResult>({beginner_level_[vector_idx], in Lookup()66 do_not_trigger_in_upper_case_[vector_idx]}); in Lookup()
3237 const int vector_idx = left_leaf_rank == 1 ? 0 : 1; in ConstantPushDownBiasAdd() local3238 const int matrix_idx = 1 - vector_idx; in ConstantPushDownBiasAdd()3240 const auto& vector_prop = (*ctx.op_child_input_props)[vector_idx]; in ConstantPushDownBiasAdd()3241 const int vector_rank = vector_idx == 0 ? left_leaf_rank : right_leaf_rank; in ConstantPushDownBiasAdd()3262 input_to_swap = vector_idx; in ConstantPushDownBiasAdd()