Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcustom_gradient.py736 def transpose(*t_args, **t_kwargs):
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUFallback.cpp61 static std::optional<c10::Device> compute_target_device(std::vector<at::Tensor>& t_args, const std:… in compute_target_device()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_eager_fallback.cpp116 std::vector<at::Tensor>& t_args, in compute_target_device()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Ddataset_ops.cc1127 DataTypeVector t_args; in __anon4377504f2102() local