Home
last modified time | relevance | path

Searched defs:make_tensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dcopy_to_device_test.py210 def make_tensor(i): function
243 def make_tensor(i): function
H A Dprefetch_to_device_test.py110 def make_tensor(i): function
/aosp_15_r20/external/pytorch/torch/testing/
H A D__init__.py5 from ._creation import make_tensor as make_tensor unknown
H A D_creation.py44 def make_tensor( function
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DFunctions.cpp9 Tensor TensorMaker::make_tensor() { in make_tensor() function in at::TensorMaker
H A DTensorBody.h642 Tensor make_tensor(Args&&... args) { in make_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dexecute.py212 def make_tensor(v, arg_name): function
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.cpp31 Tensor make_tensor( in make_tensor() function
/aosp_15_r20/external/pytorch/test/
H A Dtest_type_promotion.py1121 def make_tensor(size, dtype): function