Home
last modified time | relevance | path

Searched refs:AsTensor (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dbatch_features_test.cc29 using test::AsTensor;
42 auto dense_vec = AsTensor<float>({3.0f, 7.0f}); in TEST_F()
51 auto dense_vec = AsTensor<float>({3.0f, 7.0f}, {2, 1}); in TEST_F()
60 auto dense_vec = AsTensor<float>({3.0f, 7.0f}, {1, 2}); in TEST_F()
69 auto sparse_float_feature_indices = AsTensor<int64>({0, 0, 1, 0}); in TEST_F()
70 auto sparse_float_feature_values = AsTensor<float>({3.0f, 7.0f}); in TEST_F()
71 auto sparse_float_feature_shape = AsTensor<int64>({2, 1}); in TEST_F()
82 auto sparse_float_feature_indices = AsTensor<int64>({0, 0, 1, 0}, {2, 2}); in TEST_F()
83 auto sparse_float_feature_values = AsTensor<float>({3.0f, 7.0f}, {1, 2}); in TEST_F()
84 auto sparse_float_feature_shape = AsTensor<int64>({2, 1}); in TEST_F()
[all …]
Dexamples_iterable_test.cc41 auto dense_float_tensor = test::AsTensor<float>( in TEST_F()
44 test::AsTensor<int64>({0, 0, 2, 0, 3, 0, 4, 0}, {4, 2}); in TEST_F()
45 auto sparse_float_values1 = test::AsTensor<float>({-3.0f, 0.0f, 5.0f, 0.0f}); in TEST_F()
51 auto sparse_float_indices2 = test::AsTensor<int64>( in TEST_F()
54 test::AsTensor<float>({1.f, 4.0f, 3.f, 7.0f, 4.3f, 9.0f, 0.8f, -4.0f}); in TEST_F()
61 test::AsTensor<int64>({0, 0, 0, 1, 1, 0, 3, 0, 3, 1, 7, 0}, {6, 2}); in TEST_F()
62 auto sparse_int_values1 = test::AsTensor<int64>({1, 8, 0, 2, 0, 5}); in TEST_F()
69 test::AsTensor<int64>({1, 0, 2, 0, 3, 0, 4, 0}, {4, 2}); in TEST_F()
70 auto sparse_int_values2 = test::AsTensor<int64>({7, 13, 4, 0}); in TEST_F()
/external/tensorflow/tensorflow/core/ops/
Dmath_grad_test.cc408 auto x = test::AsTensor<float>({-3.f}); in TEST_F()
409 auto dx = test::AsTensor<float>({3.f}); in TEST_F()
416 auto x = test::AsTensor<float>({-3.f, -2.f, -1.f, 1.f, 2.f, 3.f}, in TEST_F()
419 auto dx = test::AsTensor<float>( in TEST_F()
426 auto x = test::AsTensor<float>({-3.f, -2.f, -1.f, 1.f, 2.f, 3.f}, in TEST_F()
429 auto dx = test::AsTensor<float>( in TEST_F()
436 auto x = test::AsTensor<float>({-3.f, -2.f, -1.f, 1.f, 2.f, 3.f}, in TEST_F()
439 auto dx = test::AsTensor<float>( in TEST_F()
446 auto x = test::AsTensor<float>({-3.f, -2.f, -1.f, 1.f, 2.f, 3.f}, in TEST_F()
449 auto dx = test::AsTensor<float>( in TEST_F()
[all …]
Darray_grad_test.cc72 test::AsTensor<float>({0., 1., 2., 3., 4., 5.}, {2, 3})); in TEST()
74 test::AsTensor<float>({6., 7., 8., 9., 10., 11.}, {2, 3})); in TEST()
112 test::ExpectClose(dx[0], test::AsTensor<float>({0., 1., 2., 3., 4., 5., 100., in TEST()
178 test::AsTensor<float>({0., 1., 2., 3., 4., 5., 6., 7., 8., 9., in TEST()
182 test::ExpectClose(dx[2], test::AsTensor<float>({15., 16., 17., 18., 19., 35., in TEST()
191 test::AsTensor<float>({0., 1., 2., 3., 4., 5., 6., 7., 8., 9., in TEST()
195 test::ExpectClose(dx[1], test::AsTensor<float>({15., 16., 17., 18., 19., 35., in TEST()
204 test::AsTensor<float>({0., 1., 2., 3., 4., 5., 6., 7., 8., 9., in TEST()
208 test::ExpectClose(dx[1], test::AsTensor<float>({15., 16., 17., 18., 19., 35., in TEST()
281 auto expected_dx = test::AsTensor<float>( in TEST()
[all …]
Darray_ops_test.cc282 Tensor in_t = test::AsTensor<int32>({1, 2, 3, 4}); in TEST()
650 dim_t = test::AsTensor<int32>(dims); in TEST()
656 dim_t = test::AsTensor<int32>(dims); in TEST()
871 Tensor new_shape = test::AsTensor<int32>({1, 2, 3}); in TEST()
882 new_shape = test::AsTensor<int32>({-1}); in TEST()
886 new_shape = test::AsTensor<int32>({2, -1}); in TEST()
893 new_shape = test::AsTensor<int32>({-1, -1, 2}); in TEST()
897 new_shape = test::AsTensor<int32>({-1, 2, 3}); in TEST()
901 new_shape = test::AsTensor<int32>({}); in TEST()
908 new_shape = test::AsTensor<int32>({-1}); in TEST()
[all …]
Dspectral_ops_test.cc75 Tensor fft_length = test::AsTensor<int32>({10}); in TEST()
89 fft_length = test::AsTensor<int32>({11}); in TEST()
100 fft_length = test::AsTensor<int32>({12}); in TEST()
135 Tensor fft_length = test::AsTensor<int32>({9, 10}); in TEST()
149 fft_length = test::AsTensor<int32>({10, 11}); in TEST()
160 fft_length = test::AsTensor<int32>({11, 12}); in TEST()
195 Tensor fft_length = test::AsTensor<int32>({10, 11, 12}); in TEST()
209 fft_length = test::AsTensor<int32>({11, 12, 13}); in TEST()
220 fft_length = test::AsTensor<int32>({12, 13, 14}); in TEST()
Dimage_ops_test.cc47 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST()
204 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST()
232 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST()
258 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST()
274 Tensor image_size = test::AsTensor<int32>({10, 20, 30, 40}); in TEST()
292 Tensor size = test::AsTensor<int64>({10, 20}); in TEST()
317 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST()
Drandom_ops_test.cc36 num_samples = test::AsTensor<int32>({1, 2, 3}); in TEST()
48 Tensor shape = test::AsTensor<int32>({1, 2, 3}); in TEST()
63 Tensor shape = test::AsTensor<int32>({1, 2, 3}); in TEST()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils_test.cc23 Tensor two_dim = test::AsTensor<float>({1, 3, 5, 7, 2, 4, 6, 8}, {2, 4}); in TEST()
26 Tensor multi_dim_single_value = test::AsTensor<int>({42}, {1, 1, 1, 1}); in TEST()
29 Tensor nothing = test::AsTensor<float>({}, {0}); in TEST()
34 Tensor vals = test::AsTensor<float>({2, 2, 2}, {3}); in TEST()
37 Tensor zero = test::AsTensor<float>({0}, {1}); in TEST()
43 Tensor sums = test::AsTensor<float>({9, 6}, {2}); in TEST()
44 Tensor squares = test::AsTensor<float>({29, 12}, {2}); in TEST()
50 Tensor zero = test::AsTensor<float>({0}, {1}); in TEST()
57 Tensor t = test::AsTensor<float>({0, 0, 0, 0, 0, 0, 0, 0}, {4, 2}); in TEST()
66 Tensor nothing = test::AsTensor<float>({}, {0}); in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dragged_range_op_test.cc57 test::AsTensor<int64>({0, 4, 6, 6, 10})); in TEST_F()
60 test::AsTensor<int>({0, 2, 4, 6, 5, 6, 5, 4, 3, 2})); in TEST_F()
72 test::AsTensor<int64>({0, 4, 6, 6, 10})); in TEST_F()
75 test::AsTensor<float>({0, 2, 4, 6, 5, 6, 5, 4, 3, 2}), 0.1); in TEST_F()
87 test::AsTensor<int64>({0, 8, 10, 10})); in TEST_F()
90 test::AsTensor<int>({0, 1, 2, 3, 4, 5, 6, 7, 5, 6})); in TEST_F()
102 test::AsTensor<int64>({0, 3, 3, 5})); in TEST_F()
104 test::AsTensor<int>({0, 1, 2, 0, 1})); in TEST_F()
116 test::AsTensor<int64>({0, 4, 7, 10})); in TEST_F()
119 test::AsTensor<int>({0, 3, 6, 9, 0, 4, 8, 0, 5, 10})); in TEST_F()
[all …]
Dragged_tensor_to_sparse_kernel_test.cc67 test::AsTensor<int64>({0, 0, 0, 1, 0, 2, 2, 0, 2, 1, 3, 0}, {6, 2})); in TEST_F()
69 test::AsTensor<int>({1, 2, 3, 4, 5, 6})); in TEST_F()
71 test::AsTensor<int64>({4, 3})); in TEST_F()
83 test::AsTensor<int64>({1, 0, 1, 1, 1, 2, 1, 3, 3, 0, 3, 1}, {6, 2})); in TEST_F()
85 test::AsTensor<int>({1, 2, 3, 4, 5, 6})); in TEST_F()
87 test::AsTensor<int64>({5, 4})); in TEST_F()
103 test::AsTensor<int64>(expected_splits_12_3, {12, 3})); in TEST_F()
105 test::AsTensor<int>(expected_values)); in TEST_F()
107 test::AsTensor<int64>({4, 3, 2})); in TEST_F()
129 test::AsTensor<int>(expected_values)); in TEST_F()
[all …]
Dragged_gather_op_test.cc78 test::AsTensor<int64>({0, 4, 4, 7, 9})); in TEST_F()
81 test::AsTensor<float>({.4, .5, .6, .7, .1, .2, .3, .8, .9}), 0.1); in TEST_F()
100 test::AsTensor<int64>({0, 0, 2, 3, 3, 5})); in TEST_F()
102 test::AsTensor<int64>({0, 2, 3, 3, 5, 8})); in TEST_F()
104 *GetOutput(2), test::AsTensor<float>({.1, .2, .3, .4, .5, .6, .7, .8}), in TEST_F()
127 test::AsTensor<int64>({0, 0, 2, 3, 3})); in TEST_F()
129 test::AsTensor<int64>({0, 3, 4, 4})); in TEST_F()
132 test::AsTensor<int32>({1, 2, 3, 4, 5, 6, 7, 8}, TensorShape({4, 2}))); in TEST_F()
151 test::AsTensor<int64>({0, 2, 4})); in TEST_F()
153 test::AsTensor<int64>({0, 4, 4, 7, 9})); in TEST_F()
[all …]
Dstrided_slice_op_test.cc102 Tensor begin = test::AsTensor<int32>({10, 10}); in BM_ValidateStridedSliceOp()
103 Tensor end = test::AsTensor<int32>({10 + kDim, 10 + size}); in BM_ValidateStridedSliceOp()
104 Tensor strides = test::AsTensor<int32>({1, 1}); in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsignature_test.cc302 input_tensor_ = test::AsTensor<int>({99}); in TEST_F()
303 session_.outputs = {test::AsTensor<int>({3}), test::AsTensor<int>({2})}; in TEST_F()
309 test::ExpectTensorEqual<int>(test::AsTensor<int>({3}), classes_tensor_); in TEST_F()
310 test::ExpectTensorEqual<int>(test::AsTensor<int>({2}), scores_tensor_); in TEST_F()
315 test::ExpectTensorEqual<int>(test::AsTensor<int>({99}), in TEST_F()
324 input_tensor_ = test::AsTensor<int>({99}); in TEST_F()
325 session_.outputs = {test::AsTensor<int>({3})}; in TEST_F()
331 test::ExpectTensorEqual<int>(test::AsTensor<int>({3}), classes_tensor_); in TEST_F()
336 test::ExpectTensorEqual<int>(test::AsTensor<int>({99}), in TEST_F()
344 input_tensor_ = test::AsTensor<int>({99}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc41 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({1, 1}, {1, 2})); in TEST()
53 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({42}, {})); in TEST()
64 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({3, 3}, {2})); in TEST()
69 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({31, 42}, {2})); in TEST()
83 test::AsTensor<int>({4, 6}, {2})); in TEST()
89 test::AsTensor<int>({3, 8}, {2})); in TEST()
95 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({-1, 2}, {2})); in TEST()
113 callable, {test::AsTensor<int>({1}, {}), test::AsTensor<int>({41}, {})}, in TEST()
115 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({42}, {})); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc275 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
279 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({2, 4, 6, 8})); in TEST_F()
294 y, test::AsTensor<string>({"/job:a/replica:0/task:0/device:CPU:0"}, in TEST_F()
301 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
311 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({2, 4, 6, 8})); in TEST_F()
314 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({4, 8, 12, 16})); in TEST_F()
329 y, test::AsTensor<string>({"/job:a/replica:0/task:0/device:CPU:1"}, in TEST_F()
333 y, test::AsTensor<string>({"/job:a/replica:0/task:0/device:CPU:1"}, in TEST_F()
349 y, test::AsTensor<string>({"/job:a/replica:0/task:0/device:CPU:0"}, in TEST_F()
355 y, test::AsTensor<string>({"/job:a/replica:0/task:0/device:CPU:1"}, in TEST_F()
[all …]
Dfunction_test.cc127 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
130 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({2, 4, 6, 8})); in TEST_F()
135 auto w = test::AsTensor<float>({1., 2., 3., 4.}, {2, 2}); in TEST_F()
136 auto x = test::AsTensor<float>({1., 3., 2., 4.}, {2, 2}); in TEST_F()
137 auto b = test::AsTensor<float>({0.5, 2.5}, {2}); in TEST_F()
141 y, test::AsTensor<float>({5.5, 13.5, 11.5, 27.5}, {2, 2})); in TEST_F()
379 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
383 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({2, 4, 6, 8})); in TEST_F()
386 test::ExpectTensorEqual<float>(y, test::AsTensor<float>({2, 4, 6, 8})); in TEST_F()
392 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/
Ddecision_tree_test.cc38 auto dense_float_matrix = test::AsTensor<float>({7.0f, -2.0f}, {2, 1}); in DecisionTreeTest()
39 auto sparse_float_indices1 = test::AsTensor<int64>({0, 0}, {1, 2}); in DecisionTreeTest()
40 auto sparse_float_values1 = test::AsTensor<float>({-3.0f}); in DecisionTreeTest()
41 auto sparse_float_shape1 = test::AsTensor<int64>({2, 1}); in DecisionTreeTest()
42 auto sparse_float_indices2 = test::AsTensor<int64>({1, 0}, {1, 2}); in DecisionTreeTest()
43 auto sparse_float_values2 = test::AsTensor<float>({4.0f}); in DecisionTreeTest()
44 auto sparse_float_shape2 = test::AsTensor<int64>({2, 1}); in DecisionTreeTest()
45 auto sparse_int_indices1 = test::AsTensor<int64>({0, 0}, {1, 2}); in DecisionTreeTest()
46 auto sparse_int_values1 = test::AsTensor<int64>({3}); in DecisionTreeTest()
47 auto sparse_int_shape1 = test::AsTensor<int64>({2, 1}); in DecisionTreeTest()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/
Dnode-stats_test.cc21 using tensorflow::test::AsTensor;
122 Tensor g = AsTensor<float>({0.5, 0.33, -9, 1}, g_shape); in TEST()
124 Tensor h = AsTensor<float>({3, 5, 7, 8, 5, 4, 1, 5, 7, 1, 8, 4, 8, 5, 4, 9}, in TEST()
149 Tensor g = AsTensor<float>({0.5, 0.33, -9, 1}, g_shape); in TEST()
155 h = AsTensor<float>({3, 0, 0, 0, 0, 4, 0, 0, 0, 0, 8, 0, 0, 0, 0, 9}, in TEST()
164 h = AsTensor<float>({3, 4, 8, 9}, hessian_shape); in TEST()
185 Tensor g = AsTensor<float>({0.5, 0.33, -9, 1}, g_shape); in TEST()
191 h = AsTensor<float>({3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 0, 0, 0, 0, 0}, in TEST()
200 h = AsTensor<float>({3, 0, 8, 0}, hessian_shape); in TEST()
/external/tensorflow/tensorflow/cc/framework/
Dcc_ops_test.cc54 test::ExpectTensorEqual<int>(out, test::AsTensor<int>({42}, {1, 1})); in TEST()
63 test::ExpectTensorEqual<int>(out, test::AsTensor<int>({42}, {1, 1})); in TEST()
73 test::ExpectTensorEqual<int>(out, test::AsTensor<int>({1, 2}, {2, 1})); in TEST()
84 test::ExpectClose(out, test::AsTensor<float>({-0.3, 34.2}, {1, 2})); in TEST()
92 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({1, 2, 4, 3})); in TEST()
94 test::AsTensor<int>({0, 1, 1, 2, 3, 1})); in TEST()
116 out, test::AsTensor<float>({0.98058069, -0.19611613}, {2, 1}), 1e-5); in TEST()
142 test::ExpectTensorNear<float>(out[0], test::AsTensor<float>({42.0f}, {}), in TEST()
150 test::ExpectTensorNear<float>(out[0], test::AsTensor<float>({1.0f}, {}), in TEST()
152 test::ExpectTensorEqual<bool>(out[1], test::AsTensor<bool>({false}, {})); in TEST()
[all …]
Dgradients_test.cc341 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({40}, {1, 1})); in TEST_F()
342 test::ExpectTensorEqual<int>(outputs[1], test::AsTensor<int>({20}, {1, 1})); in TEST_F()
369 test::AsTensor<int>({40, 41, 42, 43, 44, 45, 46, 47}, {4, 2})); in TEST_F()
372 test::AsTensor<int>({50, 51, 52, 53, 55, 55, 56, 57}, {4, 2})); in TEST_F()
375 test::AsTensor<int>({60, 61, 62, 63, 66, 66, 66, 67}, {4, 2})); in TEST_F()
404 outputs[0], test::AsTensor<double>({2.5, 3.5, 4.5}, {3, 1}), 1e-5); in TEST_F()
440 outputs[0], test::AsTensor<double>({17.5, 24.7, 26.8}, {3, 1}), 1e-5); in TEST_F()
556 test::AsTensor<int>({15, 18, 21, 24}, {2, 2})); in TEST_F()
561 test::AsTensor<int>({14, 16, 18, 20}, {2, 2})); in TEST_F()
566 test::AsTensor<int>({10, 12, 14, 16}, {2, 2})); in TEST_F()
[all …]
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc135 test::AsTensor<float>({-0.9f, -0.7f, -0.5f, -0.3f, -0.1f, 0.1f, 0.3f, in TEST_F()
146 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
158 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
170 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
181 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
183 Tensor features = test::AsTensor<float>( in TEST_F()
194 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
204 Tensor x_init_value = test::AsTensor<float>( in TEST_F()
230 Tensor filter = test::AsTensor<float>({0.5f}, {1, 1, 1, 1}); in TEST_F()
254 Tensor ksize = test::AsTensor<int>({1, 2, 2, 1}, {4}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc316 auto x = test::AsTensor<int32>({1, 2, 3, 4}); in TEST_F()
319 test::ExpectTensorEqual<int32>(y, test::AsTensor<int32>({2, 4, 6, 8})); in TEST_F()
330 auto x1 = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F()
331 auto x2 = test::AsTensor<float>({4, 3, 2, 1}); in TEST_F()
334 test::ExpectTensorEqual<float>(y1, test::AsTensor<float>({4, 3, 2, 1})); in TEST_F()
335 test::ExpectTensorEqual<float>(y2, test::AsTensor<float>({1, 2, 3, 4})); in TEST_F()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees_test.cc32 using test::AsTensor;
47 auto dense_matrix = test::AsTensor<float>({7.0f, -2.0f}, {2, 1}); in MultipleAdditiveTreesTest()
58 auto output_tensor = AsTensor<float>({9.0f, 23.0f}, {2, 1}); in TEST_F()
94 auto output_tensor = AsTensor<float>({0.0f, 0.0f}, {2, 1}); in TEST_F()
191 auto output_tensor = AsTensor<float>({0.0f, 0.0f, 0.0f, 0.0f}, {2, 2}); in TEST_F()
278 AsTensor<float>({0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f}, {2, 3}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dliteral_util_test.cc42 test::AsTensor<int64>(int64_values)); in TEST()
66 test::AsTensor<int_type>(int_values)); in TYPED_TEST()
74 test::AsTensor<qint_type>(qint_values)); in TYPED_TEST()

123