| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_with_tracking_alloc_test.cc | 56 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
57 test::FillValues<float>(&x_tensor, {1, 1}); in TEST()
58 Node* x = test::graph::Constant(&graph, x_tensor); in TEST()
174 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TestHWAccelerator() local
175 test::FillValues<float>(&x_tensor, {1, 1}); in TestHWAccelerator()
176 Node* x = test::graph::Constant(&graph, x_tensor); in TestHWAccelerator()
265 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
266 test::FillValues<float>(&x_tensor, {1, 1}); in TEST()
267 Node* x = test::graph::Constant(&graph, x_tensor); in TEST()
327 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
[all …]
|
| D | direct_session_test.cc | 93 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
94 test::FillValues<float>(&x_tensor, {1, 1}); in Initialize()
95 Node* x = test::graph::Constant(&graph, x_tensor); in Initialize()
535 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
536 x_tensor.flat<float>().setRandom(); in TEST_F()
537 Node* x = test::graph::Constant(&graph, x_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | gru_ops.cc | 34 const Tensor* x_tensor = nullptr; in Compute() local
35 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
52 const int64 batch_size = x_tensor->dim_size(0); in Compute()
53 const int64 input_size = x_tensor->dim_size(1); in Compute()
152 ctx, device, x_tensor->matrix<T>(), h_prev_tensor->matrix<T>(), in Compute()
177 const Tensor* x_tensor = nullptr; in Compute() local
178 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
207 const int64 batch_size = x_tensor->dim_size(0); in Compute()
208 const int64 input_size = x_tensor->dim_size(1); in Compute()
360 ctx, device, x_tensor->matrix<T>(), h_prev_tensor->matrix<T>(), in Compute()
|
| D | lstm_ops.cc | 244 const Tensor* x_tensor = nullptr; in Compute() local
245 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
268 const int64 batch_size = x_tensor->dim_size(0); in Compute()
269 const int64 input_size = x_tensor->dim_size(1); in Compute()
359 x_tensor->matrix<T>(), cs_prev_tensor->matrix<T>(), in Compute()
425 const Tensor* x_tensor = nullptr; in Compute() local
426 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
473 const int64 batch_size = x_tensor->dim_size(0); in Compute()
474 const int64 input_size = x_tensor->dim_size(1); in Compute()
643 ctx, device, use_peephole_, x_tensor->matrix<T>(), in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_gradients.py | 308 def gradient_tensor(self, x_tensor): argument
324 x_tensor_name = self._get_tensor_name(x_tensor)
372 def gradient_values_from_dump(grad_debugger, x_tensor, dump): argument
410 gradient_tensor = grad_debugger.gradient_tensor(x_tensor)
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | device_tracer_test.cc | 63 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
64 test::FillValues<float>(&x_tensor, {1, 1}); in Initialize()
65 Node* x = test::graph::Constant(&graph, x_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | listdiff_op_test.py | 37 x_tensor = ops.convert_to_tensor(x, dtype=dtype)
41 x_tensor, y_tensor, out_idx=index_dtype)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | listdiff_op_test.py | 46 x_tensor = ops.convert_to_tensor(x, dtype=dtype)
48 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | mixture_test.py | 723 x_tensor = array_ops.placeholder(shape=(), dtype=dtypes.float32)
732 x_cdf_tf = mixture_tf.cdf(x_tensor)
733 x_log_cdf_tf = mixture_tf.log_cdf(x_tensor)
738 [x_cdf_tf, x_log_cdf_tf], feed_dict={x_tensor: x_feed})
772 x_tensor = array_ops.placeholder(shape=psize, dtype=dtypes.float32)
778 x_cdf_tf = mixture_tf.cdf(x_tensor)
779 x_log_cdf_tf = mixture_tf.log_cdf(x_tensor)
785 feed_dict={x_tensor: x_feed})
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 373 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
374 test::FillValues<float>(&x_tensor, {0, 0}); in TEST_F()
375 Node* x_node = test::graph::Constant(&graph, x_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 1506 x_tensor = array_ops.placeholder(x.dtype, shape=[None] * x.ndim)
1507 feed_dict = {x_tensor: x}
1509 x_tensor = x
1512 y = image_ops.crop_to_bounding_box(x_tensor, offset_height, offset_width,
1831 x_tensor = array_ops.placeholder(x.dtype, shape=[None] * x.ndim)
1832 feed_dict = {x_tensor: x}
1834 x_tensor = x
1837 y = image_ops.pad_to_bounding_box(x_tensor, offset_height, offset_width,
2639 x_tensor = array_ops.placeholder(x.dtype, shape=[None] * x.ndim)
2640 feed_dict = {x_tensor: x}
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 302 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in FindMaxEigen() local
303 test::FillValues<float>(&x_tensor, {0, 0}); in FindMaxEigen()
304 Node* x = test::graph::Constant(&graph, x_tensor); in FindMaxEigen()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 691 def _map_fn(x_tensor): argument
695 _map_py_func, [x_tensor], [dtypes.int64, dtypes.float64])
|