Searched refs:tensorOrConstant (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| D | pointwise.cpp | 16 tensorOrConstant(inputValues[0], indices)}; in computeSign()
40 tensorOrConstant(inputValues[0], indices)}; in computeOneOperand()
62 tensorOrConstant(inputValues[0], indices), in computeTwoOperand()
63 tensorOrConstant(inputValues[1], indices), in computeTwoOperand()
87 tensorOrConstant(inputValues[0], indices), in computeTwoOperandWithAlpha()
88 tensorOrConstant(inputValues[1], indices), in computeTwoOperandWithAlpha()
89 tensorOrConstant(inputValues[2], indices), in computeTwoOperandWithAlpha()
114 tensorOrConstant(inputValues[1], indices), in computeConditionWithTwoOperand()
115 tensorOrConstant(inputValues[2], indices), in computeConditionWithTwoOperand()
121 inputs.begin(), tensorOrConstant(inputValues[0], indices)); in computeConditionWithTwoOperand()
[all …]
|
| D | norm.cpp | 37 tensorOrConstant(inputs[0], indices), // input in computeBatchNorm()
38 tensorOrConstant(inputs[3], {c}), // mean in computeBatchNorm()
39 tensorOrConstant(inputs[4], {c}), // var in computeBatchNorm()
46 weight = tensorOrConstant(inputs[1], {c}); in computeBatchNorm()
50 bias = tensorOrConstant(inputs[2], {c}); in computeBatchNorm()
|
| D | softmax.cpp | 107 return tensorOrConstant( in computeSoftmax()
116 auto inp = tensorOrConstant( in computeSoftmax()
150 auto inp = tensorOrConstant( in computeSoftmax()
|
| D | misc.h | 36 ExprHandle tensorOrConstant(
|
| D | quantization.cpp | 165 quant(tensorOrConstant(inputs[0], indices), dtype, qscale, qzero); in computeQuantizePerTensor()
201 auto lhs = tensorOrConstant(inputs[0], indices); in computeQuantizedAdd()
202 auto rhs = tensorOrConstant(inputs[1], indices); in computeQuantizedAdd()
662 auto y = dequant(tensorOrConstant(inputs[0], indices), dtype, qscale, qzero); in computeDequantize()
|
| D | misc.cpp | 256 ExprHandle tensorOrConstant( in tensorOrConstant() function
660 tensorOrConstant(nonEmptyInputs[0], newAxes), highType); in computeCat()
669 promoteToDtype(tensorOrConstant(input, newAxes), highType)); in computeCat()
|
| D | reduction.cpp | 95 auto indexed = tensorOrConstant(inputs[0], indices_exprs); in computeSum()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | kernel.h | 44 ExprHandle tensorOrConstant(
|
| D | lowerings.cpp | 1109 tensorOrConstant(inputs[0], indices), in nnc_lowerings_lazy_registration()
1110 tensorOrConstant(inputs[1], indices), in nnc_lowerings_lazy_registration()
|