| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | prepare_binary.cpp | 59 node->replaceInput(1, unsqueezed); in handleBinaryOpInputs()
87 node->replaceInput(1, to_node_output); in handleBinaryOpInputs()
94 node->replaceInput(0, to_node_output); in handleBinaryOpInputs()
142 node->replaceInput(1, mul); in mayDecomposeAdd()
144 node->replaceInput(2, one); in mayDecomposeAdd()
|
| D | graph_helper.cpp | 21 node->replaceInput(2, v); in fixConvOptionalBias()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | prepare_division_for_onnx.cpp | 33 it->replaceInput(0, floattensor_inputs[0]); in PrepareDivisionForONNXOnBlock()
34 it->replaceInput(1, floattensor_inputs[1]); in PrepareDivisionForONNXOnBlock()
|
| D | peephole.cpp | 174 n->replaceInput(position, unexpanded_input); in fuseBroadcast()
195 n->replaceInput(0, origInput->node()->input()); in fuseConsecutiveTransposes()
233 n->replaceInput(i, inp->node()->input()); in fuseTransposeIntoGemm()
340 user->replaceInput(1, gather_indices->output()); in pushPackingPastRnn()
531 n->replaceInput(input_index, fixed_init_state->outputs()[0]); in fixDefaultRNNState()
862 origLogSoftmaxNode->replaceInput(0, transpose->inputs().at(0)); in fuseLogSoftmaxNllLoss()
890 origLogSoftmaxNode->replaceInput(0, transpose->inputs().at(0)); in fuseLogSoftmaxNllLoss()
893 origNllLossNode->replaceInput(1, reshape->inputs().at(0)); in fuseLogSoftmaxNllLoss()
|
| D | remove_inplace_ops_for_onnx.cpp | 360 node->replaceInput(index, newNode->output()); in PrepareForRemoveMutations()
579 n->replaceInput(i, alias); in correctAliasReferences()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | loop_unrolling.cpp | 196 loop_epilogue->replaceInput(i + 2, loop->outputs()[i]); in unroll()
207 loop->replaceInput(0, unrolled_iter_count); in unroll()
208 loop_epilogue->replaceInput( in unroll()
354 n->replaceInput(LOOP_DEPS_WITH_COND_OFFSET + i, peeled_copy->output(i)); in PeelLoop()
368 adjusted_iter_counter->node()->replaceInput( in PeelLoop()
|
| D | fixup_trace_scope_blocks.cpp | 173 n->replaceInput(inp_idx, local_remaps[inp]); in convertAttrReferencesToLocalGetAttrs()
211 n->replaceInput(inp_idx, replaced_value); in replaceTracedAttrInputOnNode()
219 n->replaceInput(inp_idx, remapped); in replaceTracedAttrInputOnNode()
263 n->replaceInput(i, remap[inp_remapped]); in processNode()
294 n->replaceInput(i, remap[inp]); in processNode()
|
| D | frozen_conv_folding.cpp | 274 add_or_sub->replaceInput( in FoldFrozenConvAddOrSub()
345 mul_or_div->replaceInput(1, b->owningGraph()->insertConstant(mul_tensor)); in FoldFrozenConvMulOrDiv()
367 mul_or_div->replaceInput(0, b->owningGraph()->insertConstant(bias)); in FoldFrozenConvMulOrDiv()
368 mul_or_div->replaceInput( in FoldFrozenConvMulOrDiv()
|
| D | normalize_ops.cpp | 28 newSub->replaceInput(0, args[1]); in normalizeRSub()
29 newSub->replaceInput(1, args[0]); in normalizeRSub()
|
| D | peephole_non_tensor.cpp | 85 node.replaceInput(0, dep.inputs()[0]); in trySimplifyAddOrSub()
86 node.replaceInput(1, node.owningGraph()->insertConstant(merged)); in trySimplifyAddOrSub()
|
| D | hoist_conv_packed_params.cpp | 76 getConvPackedParamsNode->replaceInput(0, rootModuleAsValue); in hoistConvPackedParams()
|
| D | remove_inplace_ops.cpp | 137 it->replaceInput(1, newInputNode->outputs().at(0)); in ImplicitCastForBinaryInplaceOps()
|
| D | frozen_ops_to_mkldnn.cpp | 731 subgraph_node->replaceInput(i, to_mkldnn->output()); in ComputeSubgraphInMKLDNN()
763 body_node->replaceInput(0, node->outputs().at(0)); in ComputeSubgraphInMKLDNN()
764 body_node->replaceInput(1, node->outputs().at(1)); in ComputeSubgraphInMKLDNN()
|
| D | remove_mutation.cpp | 203 list_construct->replaceInput(pos, node->input(2)); in RemoveListMutation()
|
| D | fold_conv_bn.cpp | 85 conv_node->replaceInput(conv_bias_index, bias_attr_val); in replaceConvBiasWithGetAttr()
|
| D | guard_elimination.cpp | 155 use.user->replaceInput(use.offset, guard_output); in removeDominatedGuards()
|
| D | peephole.cpp | 81 u.user->replaceInput(0, node->inputs().at(0)); in optimizeBlock()
|
| D | specialize_autogradzero.cpp | 299 ret->replaceInput(i, ngo); in guardSpecializations()
|
| /external/pytorch/torch/csrc/jit/ir/ |
| D | ir_views.h | 105 node_->replaceInput(0, new_max_trip_count); in replaceMaxTripCount()
108 node_->replaceInput(1, new_input_condition); in replaceInputCondition()
|
| D | ir.h | 586 Value* replaceInput(size_t i, Value* newValue);
1111 output_->replaceInput(i, n); in replaceOutput()
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | register_packed_params.cpp | 116 use.user->replaceInput(use.offset, packed_param_attr); in RegisterPrePackParams()
|
| D | insert_quant_dequant.cpp | 150 user->replaceInput(uses[i].offset, dequant->output()); in insertDeQuantForAllUse()
635 user->replaceInput(use.offset, cloned_node->output()); in ReplicateClampScalarArgs()
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | profiling_record.cpp | 205 n->replaceInput(offset, pn->output()); in insertShapeProfile()
|
| D | autodiff.cpp | 790 new_vjp->node()->replaceInput(1, tmp_vjp_prev); in lambdaLiftReverse()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/ |
| D | pattern_conversion.cpp | 311 last_node->replaceInput(0, old_node->input(0)); in ConvertIndexPutToONNX()
|