Searched refs:batched_tensor_inputs_position (Results 1 – 2 of 2) sorted by relevance
117 VmapDimVector batched_tensor_inputs_position; in batchedTensorInplaceForLoopFallback() local162 batched_tensor_inputs_position.push_back(static_cast<int64_t>(idx)); in batchedTensorInplaceForLoopFallback()187 auto batched_tensor_inputs_pos_iter = batched_tensor_inputs_position.begin(); in batchedTensorInplaceForLoopFallback()193 if (batched_tensor_inputs_pos_iter == batched_tensor_inputs_position.end() in batchedTensorInplaceForLoopFallback()292 VmapDimVector batched_tensor_inputs_position; in batchedTensorForLoopFallback() local307 batched_tensor_inputs_position.push_back(static_cast<int64_t>(idx)); in batchedTensorForLoopFallback()341 auto batched_tensor_inputs_pos_iter = batched_tensor_inputs_position.begin(); in batchedTensorForLoopFallback()347 if (batched_tensor_inputs_pos_iter == batched_tensor_inputs_position.end() in batchedTensorForLoopFallback()433 VmapDimVector batched_tensor_inputs_position; in batchedNestedTensorForLoopFallback() local448 batched_tensor_inputs_position.push_back(static_cast<int64_t>(idx)); in batchedNestedTensorForLoopFallback()[all …]
112 VmapDimVector batched_tensor_inputs_position; in batchedTensorInplaceForLoopFallback() local157 batched_tensor_inputs_position.push_back(idx); in batchedTensorInplaceForLoopFallback()182 auto batched_tensor_inputs_pos_iter = batched_tensor_inputs_position.begin(); in batchedTensorInplaceForLoopFallback()188 if (batched_tensor_inputs_pos_iter == batched_tensor_inputs_position.end() in batchedTensorInplaceForLoopFallback()276 VmapDimVector batched_tensor_inputs_position; in batchedTensorForLoopFallback() local291 batched_tensor_inputs_position.push_back(idx); in batchedTensorForLoopFallback()325 auto batched_tensor_inputs_pos_iter = batched_tensor_inputs_position.begin(); in batchedTensorForLoopFallback()331 if (batched_tensor_inputs_pos_iter == batched_tensor_inputs_position.end() in batchedTensorForLoopFallback()