Home
last modified time | relevance | path

Searched defs:broadcast_tensors (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dloss.h118 std::vector<torch::Tensor> broadcast_tensors = in mse_loss() local
/aosp_15_r20/external/pytorch/torch/
H A Dfunctional.py47 def broadcast_tensors(*tensors): function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_fuser.cpp587 at::ArrayRef<Value*> broadcast_tensors(value_list inputs) { in broadcast_tensors() function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py2698 def broadcast_tensors(*tensors) -> List[TensorLikeType]: function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py810 def broadcast_tensors(*inputs): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp558 std::vector<Tensor> broadcast_tensors(TensorList tensors) { in broadcast_tensors() function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py6228 def broadcast_tensors(g: jit_utils.GraphContext, self): function