Home
last modified time | relevance | path

Searched refs:dtypeToCppString (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dir_printer.cpp13 std::string IRPrinter::dtypeToCppString(const Dtype& dtype) { in dtypeToCppString() function in torch::jit::tensorexpr::IRPrinter
230 os() << dtypeToCppString(dtype) << "("; in visit()
249 os() << "(dtype=" << dtypeToCppString(dtype); in visit()
437 os() << "for (" << dtypeToCppString(var->dtype()) << " " << vv << " = " in visit()
466 << "); // dtype=" << dtypeToCppString(v->dtype()); in visit()
501 os() << dtypeToCppString(v->var()->dtype()) << " " << *v->var(); in visit()
H A Dcuda_codegen.cpp70 std::string CudaPrinter::dtypeToCppString(const Dtype& dtype) { in dtypeToCppString() function in torch::jit::tensorexpr::CudaPrinter
198 os() << dtypeToCppString(alloc->dtype()) << " " << (*alloc->buffer_var()) in print_flat_alloc()
236 : ("(" + dtypeToCppString(v->dtype()) + ")"); in visit()
483 os() << dtypeToCppString(v->var()->dtype()); in visit()
924 os() << printer_->dtypeToCppString(dtype) in Initialize()
H A Dir_printer.h69 virtual std::string dtypeToCppString(const Dtype& dtype);
H A Dcuda_codegen.h183 std::string dtypeToCppString(const Dtype& dtype) override;