Home
last modified time | relevance | path

Searched defs:x_tensor (Results 1 – 11 of 11) 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
174 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TestHWAccelerator() local
256 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
318 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
Ddirect_session_test.cc103 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
682 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_gradients.py308 def gradient_tensor(self, x_tensor): argument
372 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
465 const Tensor* x_tensor = nullptr; in Compute() local
975 const Tensor x_tensor = slicer.InputSlice(*x, t, "x"); in Compute() local
1240 const Tensor& x_tensor = slicer.InputSlice(*x, t, "x"); in Compute() local
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc84 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.py777 def _map_fn(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/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1835 auto x_tensor = x_type.cast<TensorType>(); in inferReturnTypes() local