Home
last modified time | relevance | path

Searched defs:unsqueezed (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/jit/codegen/onednn/
Dprepare_binary.cpp57 auto unsqueezed = g->insert(aten::unsqueeze, {t, 0}); in handleBinaryOpInputs() local
/external/pytorch/aten/src/ATen/native/
DTensorIteratorReduce.cpp45 auto unsqueezed = dst.unsqueeze(0); in two_pass_reduction() local
/external/pytorch/torch/csrc/jit/passes/onnx/
Dhelper.cpp208 std::vector<Value*> unsqueezed; in transformToONNXConcatNode() local
/external/executorch/backends/vulkan/test/
Dvulkan_compute_api_test.cpp107 const bool unsqueezed = false) { in get_reference_strides()