Home
last modified time | relevance | path

Searched defs:tensor_output (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_dtype_analysis.py331 def assert_tensor_dtype_equal(self, tensor_output, graph_dtype): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp178 Value* tensor_output = tensor_node->output(); in insertScalarToTensor() local