Home
last modified time | relevance | path

Searched defs:tensor_inputs (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_fuser.cpp846 auto tensor_inputs = filter( in tryToMoveChunk() local
994 auto tensor_inputs = filter(n->inputs(), [](Value* v) { in buildShapeExpressions() local
H A Dtensorexpr_fuser.cpp469 auto tensor_inputs = filter(n->inputs(), [](Value* v) { in buildShapeExpressions() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_graph_builder.cpp67 std::vector<at::Tensor> tensor_inputs, in runGraphOnInputs()
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2404 Slice<TensorInfo> tensor_inputs; in getsetitem_flat() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc5743 std::vector<ITensorProxyPtr> tensor_inputs; in ConvertAddN() local