Searched refs:ExprHandleVectorToExprVector (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | ir.cpp | 55 alloc<Load>(dtype, buf.node(), ExprHandleVectorToExprVector(indices))); in make()
76 buf.node(), ExprHandleVectorToExprVector(indices), value.node()); in make()
193 buf.node(), func_name, buf_arg_nodes, ExprHandleVectorToExprVector(args)); in make()
216 ExprHandleVectorToExprVector(args)); in make()
228 std::vector<ExprPtr> ExprHandleVectorToExprVector( in ExprHandleVectorToExprVector() function
|
| H A D | expr.cpp | 442 ExprHandleVectorToExprVector(dims), in make()
445 ExprHandleVectorToExprVector(strides))); in make()
458 opt_strides = ExprHandleVectorToExprVector(*strides); in make()
462 ExprHandleVectorToExprVector(dims), in make()
|
| H A D | tensor.h | 40 ExprHandleVectorToExprVector(reduce_dims), in Tensor()
308 return ExprHandle(alloc<Load>(node(), ExprHandleVectorToExprVector(params))); in load()
314 return ExprHandle(alloc<Load>(node(), ExprHandleVectorToExprVector(params))); in load()
|
| H A D | kernel.cpp | 619 ExprHandleVectorToExprVector(biasShape), in computeValue()
1448 ExprHandleVectorToExprVector(te_sizes), in bindConstant()
|
| H A D | ir.h | 908 TORCH_API std::vector<ExprPtr> ExprHandleVectorToExprVector(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | quantization.cpp | 169 ExprHandleVectorToExprVector(outputShape), in computeQuantizePerTensor()
211 ExprHandleVectorToExprVector(outputShape), in computeQuantizedAdd()
664 "dequantize", ExprHandleVectorToExprVector(outputShape), dtype); in computeDequantize()
|
| H A D | misc.cpp | 533 auto output_sizes_expr = ExprHandleVectorToExprVector(outputShape); in computeCatWoConditionals()
534 auto output_strides_expr = ExprHandleVectorToExprVector(outputStrides); in computeCatWoConditionals()
|