Searched refs:block_output (Results 1 – 5 of 5) sorted by relevance
81 Value* block_output = if_block->outputs().at(i); in removeIfNodeOutputsWithRefinements() local82 if (!block_output->type()->cast<IntType>()) { in removeIfNodeOutputsWithRefinements()87 if (!if_node->isDominatedBy(block_output->node())) { in removeIfNodeOutputsWithRefinements()97 block_output->node()->kind() == prim::Constant) { in removeIfNodeOutputsWithRefinements()106 if (!other_block_refinements.count(block_output)) { in removeIfNodeOutputsWithRefinements()109 if (other_block_refinements.at(block_output) == *other_const_value) { in removeIfNodeOutputsWithRefinements()110 if_node->outputs().at(i)->replaceAllUsesWith(block_output); in removeIfNodeOutputsWithRefinements()
36 block_output=cli_flag)
112 block_output=True, argument133 self._block_output = block_output
211 Value* block_output = block->outputs().at(i); in ReplaceBlockOutputWithOptional() local214 block_output->replaceAllUsesAfterNodeWith(opt_node, opt_node->output()); in ReplaceBlockOutputWithOptional()215 if (!block_output->type()->cast<NoneType>()) { in ReplaceBlockOutputWithOptional()216 opt_node->addInput(block_output); in ReplaceBlockOutputWithOptional()217 opt_node->copyMetadata(block_output->node()); in ReplaceBlockOutputWithOptional()
3766 VectorMap<float> block_output(output_data + i * depth, depth, 1); in LogSoftmax()3772 block_output = block_input.array() - max - log_sum; in LogSoftmax()