Searched refs:curr_op (Results 1 – 6 of 6) sorted by relevance
34 auto curr_op = op_it->get(); in Run() local35 if (curr_op->type != OperatorType::kLstmCell) { in Run()39 const auto* curr_lstm_op = static_cast<LstmCellOperator*>(curr_op); in Run()49 *model, curr_op->inputs[LstmCellOperator::WEIGHTS_INPUT]) || in Run()51 *model, curr_op->inputs[LstmCellOperator::BIASES_INPUT])) { in Run()56 if (!model->GetArray(curr_op->outputs[LstmCellOperator::ACTIV_OUTPUT]) in Run()65 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run()71 model->GetArray(curr_op->outputs[LstmCellOperator::ACTIV_OUTPUT]) in Run()78 curr_op->inputs[LstmCellOperator::ACTIV_OUTPUT]; in Run()82 curr_op->inputs[LstmCellOperator::PREV_ACTIV_INPUT]; in Run()[all …]
81 Operation *curr_op = op; in IsOplistMatch() local85 if (!curr_op->hasOneUse()) return false; in IsOplistMatch()86 curr_op = *curr_op->getUsers().begin(); in IsOplistMatch()89 if (is_matched.contains(curr_op)) return false; in IsOplistMatch()93 if (curr_op->getName().getIdentifier() != Identifier::get(op_name, ctx)) in IsOplistMatch()97 if (curr_op->getAttr(kDeviceAttr) != device) return false; in IsOplistMatch()100 if (curr_op->getBlock() != block) return false; in IsOplistMatch()104 if (!MemoryEffectOpInterface::hasNoEffect(curr_op)) return false; in IsOplistMatch()107 matched_ops.push_back(curr_op); in IsOplistMatch()
765 Operation* curr_op = op_to_visit.op; in RewriteFunction() local769 Operation* next_op = curr_op->getNextNode(); in RewriteFunction()771 if (auto host_compute = dyn_cast<TF::_XlaHostComputeMlirOp>(curr_op)) { in RewriteFunction()773 } else if (auto send_to_host = dyn_cast<TF::XlaSendToHostOp>(curr_op)) { in RewriteFunction()775 } else if (auto recv_from_host = dyn_cast<TF::XlaRecvFromHostOp>(curr_op)) { in RewriteFunction()777 } else if (auto call = dyn_cast<mlir::CallOp>(curr_op)) { in RewriteFunction()789 } else if (auto region_if = dyn_cast<IfOp>(curr_op)) { in RewriteFunction()794 } else if (auto region_while = dyn_cast<WhileOp>(curr_op)) { in RewriteFunction()799 } else if (auto region_terminator = dyn_cast<mhlo::ReturnOp>(curr_op)) { in RewriteFunction()804 } else if (auto func_terminator = dyn_cast<mlir::ReturnOp>(curr_op)) { in RewriteFunction()
179 grpc_transport_stream_op_batch* curr_op; member805 static bool op_can_be_run(grpc_transport_stream_op_batch* curr_op, in op_can_be_run() argument932 else if (curr_op->send_initial_metadata && in op_can_be_run()936 } else if (curr_op->send_message && in op_can_be_run()940 } else if (curr_op->send_message && in op_can_be_run()944 } else if (curr_op->send_trailing_metadata && in op_can_be_run()948 } else if (curr_op->recv_initial_metadata && in op_can_be_run()952 } else if (curr_op->recv_message && in op_can_be_run()956 } else if (curr_op->cancel_stream && in op_can_be_run()960 } else if (curr_op->recv_trailing_metadata) { in op_can_be_run()[all …]
200 grpc_transport_stream_op_batch* curr_op = nullptr; member823 static bool op_can_be_run(grpc_transport_stream_op_batch* curr_op, in op_can_be_run() argument966 else if (curr_op->send_initial_metadata && in op_can_be_run()970 } else if (curr_op->send_message && in op_can_be_run()974 } else if (curr_op->send_message && in op_can_be_run()978 } else if (curr_op->send_trailing_metadata && in op_can_be_run()982 } else if (curr_op->recv_initial_metadata && in op_can_be_run()986 } else if (curr_op->recv_message && in op_can_be_run()990 } else if (curr_op->cancel_stream && in op_can_be_run()994 } else if (curr_op->recv_trailing_metadata) { in op_can_be_run()[all …]
367 curr_op = queue.popleft()368 if curr_op in visited: continue369 visited.add(curr_op)370 if curr_op in from_ops:371 target_op = curr_op373 queue.extend(t.op for t in _NonEagerInputs(curr_op, xs_set))