Home
last modified time | relevance | path

Searched refs:isPropagateQuantBinaryOp (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/quantization/
Dhelper.cpp484 bool isPropagateQuantBinaryOp(Node* n) { in isPropagateQuantBinaryOp() function
489 return isPropagateQuantSingleInputOp(n) || isPropagateQuantBinaryOp(n); in isPropagateQuantOp()
493 return isPropagateQuantBinaryOp(n) && isScalar(n->input(1)); in isBinaryOpWithScalarInput()
Dhelper.h83 TORCH_API bool isPropagateQuantBinaryOp(Node* n);
Dinsert_observers.cpp429 } else if (isPropagateQuantBinaryOp(n)) { in shouldObserve()