| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | direct_session_with_tracking_alloc_test.cc | 56 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
175 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TestHWAccelerator() local
258 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
321 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
|
| H A D | direct_session_test.cc | 104 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
683 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | atest.cpp | 22 Tensor x_tensor, y_tensor; member in atest
48 const Tensor& x_tensor, in unit_binary_ops_test()
78 const Tensor& x_tensor, in run_binary_ops_test()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/ |
| H A D | debug_gradients.py | 302 def gradient_tensor(self, x_tensor): argument
366 def gradient_values_from_dump(grad_debugger, x_tensor, dump): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | gru_ops.cc | 34 const Tensor* x_tensor = nullptr; in Compute() local
177 const Tensor* x_tensor = nullptr; in Compute() local
|
| H A D | lstm_ops.cc | 307 const Tensor* x_tensor = nullptr; in Compute() local
524 const Tensor* x_tensor = nullptr; in Compute() local
1034 const Tensor x_tensor = slicer.InputSlice(*x, t, "x"); in Compute() local
1322 const Tensor& x_tensor = slicer.InputSlice(*x, t, "x"); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | data_format_vec_permute.cc | 155 ITensorProxyPtr x_tensor = in Convert() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | device_tracer_test.cc | 91 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | master_test.cc | 374 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | run_handler_test.cc | 607 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | map_test.py | 780 def _map_fn(x_tensor): argument
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | gradcheck.py | 154 def _iter_tensor(x_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session_test.cc | 337 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in FindMaxEigen() local
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 504 def _xy_helper_corrcoef(x_tensor, y_tensor=None, rowvar=True): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 3605 def helper(x_tensor, y_func, z_func, r_func=None): argument
|