| /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()
256 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
257 test::FillValues<float>(&x_tensor, {1, 1}); in TEST()
258 Node* x = test::graph::Constant(&graph, x_tensor); in TEST()
318 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST() local
[all …]
|
| D | direct_session_test.cc | 103 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
104 test::FillValues<float>(&x_tensor, {1, 1}); in Initialize()
105 Node* x = test::graph::Constant(&graph, x_tensor); in Initialize()
682 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
683 x_tensor.flat<float>().setRandom(); in TEST_F()
684 Node* x = test::graph::Constant(&graph, x_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| 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 | 307 const Tensor* x_tensor = nullptr; in Compute() local
308 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
331 const int64 batch_size = x_tensor->dim_size(0); in Compute()
332 const int64 input_size = x_tensor->dim_size(1); in Compute()
422 x_tensor->matrix<T>(), cs_prev_tensor->matrix<T>(), in Compute()
465 const Tensor* x_tensor = nullptr; in Compute() local
466 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
513 const int64 batch_size = x_tensor->dim_size(0); in Compute()
514 const int64 input_size = x_tensor->dim_size(1); in Compute()
683 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/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/core/profiler/internal/gpu/ |
| D | device_tracer_test.cc | 84 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
85 test::FillValues<float>(&x_tensor, {1, 1}); in Initialize()
86 Node* x = test::graph::HostConstant(&graph, x_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_util_test.py | 390 x_tensor = ops.convert_to_tensor(x)
397 for inputs in [x, x_tensor, x_placeholder, x_list]:
406 for inputs in [x, x_tensor, x_placeholder, x_list]:
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 374 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in TEST_F() local
375 test::FillValues<float>(&x_tensor, {0, 0}); in TEST_F()
376 Node* x_node = test::graph::Constant(&graph, x_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | run_handler_test.cc | 607 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in Initialize() local
608 test::FillValues<float>(&x_tensor, {1, 1}); in Initialize()
609 Node* x = test::graph::Constant(&graph, x_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 1705 x_tensor = ops.convert_to_tensor(x)
1707 x_tensor = x
1709 y = image_ops.crop_to_bounding_box(x_tensor, offset_height, offset_width,
2032 x_tensor = ops.convert_to_tensor(x)
2034 x_tensor = x
2041 return self.evaluate(pad_bbox(x_tensor, offset_height, offset_width,
3066 x_tensor = ops.convert_to_tensor(x)
3069 x_tensor = x
3079 return self.evaluate(resize_func(x_tensor))
3613 x_tensor = ops.convert_to_tensor(x)
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 337 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1})); in FindMaxEigen() local
338 test::FillValues<float>(&x_tensor, {0, 0}); in FindMaxEigen()
339 Node* x = test::graph::Constant(&graph, x_tensor); in FindMaxEigen()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 777 def _map_fn(x_tensor): argument
781 _map_py_func, [x_tensor], [dtypes.int64, dtypes.float64])
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1835 auto x_tensor = x_type.cast<TensorType>(); in inferReturnTypes() local
1843 if (x_tensor.getElementType() != y_tensor.getElementType() || in inferReturnTypes()
1855 if (x_type == y_type || !x_tensor.hasRank()) { in inferReturnTypes()
|