MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / has_output

Method has_output

oneflow/core/framework/op_expr.cpp:374–378  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

372 return tuple_index >= 0;
373 }
374 bool has_output(const std::string& arg_name, int32_t index) const override {
375 const auto& arg_tuple = *user_op_expr_->output_arg_tuple();
376 int32_t tuple_index = arg_tuple.TensorTupleIndex4ArgNameAndIndex(arg_name, index);
377 return tuple_index >= 0;
378 }
379 int32_t input_size(const std::string& arg_name) const override {
380 const auto& arg_tuple = *user_op_expr_->input_arg_tuple();
381 return arg_tuple.arg_name2bn_index2tensor_tuple_index().at(arg_name).size();

Callers

nothing calls this directly

Calls 2

output_arg_tupleMethod · 0.80

Tested by

no test coverage detected