Searched refs:block_arg_num (Results 1 – 2 of 2) sorted by relevance
545 auto block_arg_num = GetBlockArgNum(input); in GetInputBlockArgNum() local546 if (block_arg_num.hasValue()) return block_arg_num; in GetInputBlockArgNum()554 auto block_arg_num = GetBlockArgNum(pad_op.input()); in GetInputBlockArgNum() local555 if (block_arg_num.hasValue()) return block_arg_num; in GetInputBlockArgNum()558 auto block_arg_num = GetBlockArgNum(cast_op.x()); in GetInputBlockArgNum() local559 if (block_arg_num.hasValue()) return block_arg_num; in GetInputBlockArgNum()
389 const int block_arg_num = arg.getArgNumber(); in Print() local390 if (block_arg_num < num_replicated_block_args) { in Print()393 std::next(op.replicated_inputs().begin(), block_arg_num * n), in Print()394 std::next(op.replicated_inputs().begin(), (block_arg_num + 1) * n)); in Print()398 block_arg_num - num_replicated_block_args)); in Print()