Home
last modified time | relevance | path

Searched defs:x_tensor (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc56 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
Ddirect_session_test.cc104 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
683 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_gradients.py302 def gradient_tensor(self, x_tensor): argument
366 def gradient_values_from_dump(grad_debugger, x_tensor, dump): argument
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.cc34 const Tensor* x_tensor = nullptr; in Compute() local
177 const Tensor* x_tensor = nullptr; in Compute() local
Dlstm_ops.cc307 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
/external/pytorch/aten/src/ATen/test/
Datest.cpp22 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()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Ddata_format_vec_permute.cc155 ITensorProxyPtr x_tensor = in Convert() local
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Ddevice_tracer_test.cc91 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc374 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc607 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py780 def _map_fn(x_tensor): argument
/external/pytorch/torch/autograd/
Dgradcheck.py154 def _iter_tensor(x_tensor): argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc337 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in FindMaxEigen() local
/external/pytorch/torch/_numpy/
D_funcs_impl.py504 def _xy_helper_corrcoef(x_tensor, y_tensor=None, rowvar=True): argument
/external/pytorch/test/
Dtest_mps.py3605 def helper(x_tensor, y_func, z_func, r_func=None): argument