| /external/tensorflow/tensorflow/core/kernels/ |
| crop_and_resize_op_test.cc | 53 AddInputFromArray<T>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); \
54 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); \
55 AddInputFromArray<int32>(TensorShape({1}), {0}); \
56 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); \
59 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1})); \
66 AddInputFromArray<T>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); \
67 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); \
68 AddInputFromArray<int32>(TensorShape({1}), {0}); \
69 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); \
72 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1}));
[all...] |
| dynamic_stitch_op_test.cc | 52 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7});
53 AddInputFromArray<int32>(TensorShape({5}), {1, 6, 2, 3, 5});
54 AddInputFromArray<float>(TensorShape({3}), {0, 40, 70});
55 AddInputFromArray<float>(TensorShape({5}), {10, 60, 20, 30, 50});
59 Tensor expected(allocator(), DT_FLOAT, TensorShape({8}));
68 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7});
69 AddInputFromArray<int32>(TensorShape({2}), {1, 6});
70 AddInputFromArray<int32>(TensorShape({3}), {2, 3, 5});
71 AddInputFromArray<float>(TensorShape({3, 2}), {0, 1, 40, 41, 70, 71});
72 AddInputFromArray<float>(TensorShape({2, 2}), {10, 11, 60, 61})
[all...] |
| resize_nearest_neighbor_op_test.cc | 77 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
78 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
81 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1}));
95 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
96 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
99 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1}));
112 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
113 AddInputFromArray<int32>(TensorShape({2}), {3, 3});
116 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 3, 3, 1}));
133 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4})
[all...] |
| roll_op_test.cc | 56 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4});
57 AddInputFromArray<int32>(TensorShape({}), {3});
58 AddInputFromArray<int32>(TensorShape({}), {0});
62 Tensor expected(allocator(), DT_FLOAT, TensorShape({5}));
71 AddInputFromArray<string>(TensorShape({5}), {"a", "b", "c", "d", "e"});
72 AddInputFromArray<int32>(TensorShape({}), {3});
73 AddInputFromArray<int32>(TensorShape({}), {0});
77 Tensor expected(allocator(), DT_STRING, TensorShape({5}));
87 TensorShape({5}), {std::complex<float>(0, 10), std::complex<float>(1, 11),
90 AddInputFromArray<int32>(TensorShape({}), {3})
[all...] |
| fused_batch_norm_op_test.cc | 43 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}),
45 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0});
46 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0});
47 AddInputFromArray<float>(TensorShape({0}), {});
48 AddInputFromArray<float>(TensorShape({0}), {});
52 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 6, 2}));
57 Tensor expected_mean(allocator(), DT_FLOAT, TensorShape({2}));
61 Tensor expected_variance(allocator(), DT_FLOAT, TensorShape({2}));
77 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}),
79 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0})
[all...] |
| requantize_op_test.cc | 54 AddInputFromArray<qint32>(TensorShape({value_count}),
56 AddInputFromArray<float>(TensorShape({1}), {-256.0f});
57 AddInputFromArray<float>(TensorShape({1}), {256.0f});
58 AddInputFromArray<float>(TensorShape({1}), {-1.0f});
59 AddInputFromArray<float>(TensorShape({1}), {1.0f});
61 Tensor expected(allocator(), DT_QUINT8, TensorShape({value_count}));
72 AddInputFromArray<qint32>(TensorShape({value_count}),
74 AddInputFromArray<float>(TensorShape({1}), {-256.0f});
75 AddInputFromArray<float>(TensorShape({1}), {256.0f});
76 AddInputFromArray<float>(TensorShape({1}), {0.01f})
[all...] |
| non_max_suppression_op_test.cc | 48 TensorShape({6, 4}),
51 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f});
52 AddInputFromArray<int>(TensorShape({}), {3});
55 Tensor expected(allocator(), DT_INT32, TensorShape({3}));
62 AddInputFromArray<float>(TensorShape({6, 4}),
65 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f});
66 AddInputFromArray<int>(TensorShape({}), {3});
69 Tensor expected(allocator(), DT_INT32, TensorShape({3}));
77 TensorShape({6, 4}),
80 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f})
[all...] |
| quantize_and_dequantize_op_test.cc | 52 AddInputFromArray<float>(TensorShape({1}), {-3.5});
53 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min
54 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max
57 Tensor expected(allocator(), DT_FLOAT, TensorShape({1}));
77 AddInputFromArray<float>(TensorShape({1}), {-3.5});
78 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min
79 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max
80 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits
83 Tensor expected(allocator(), DT_FLOAT, TensorShape({1}));
104 AddInputFromArray<float>(TensorShape({7})
[all...] |
| sequence_ops_test.cc | 59 AddInputFromArray<int32>(TensorShape({}), {0});
60 AddInputFromArray<int32>(TensorShape({}), {10});
61 AddInputFromArray<int32>(TensorShape({}), {2});
65 Tensor expected(allocator(), DT_INT32, TensorShape({5}));
74 AddInputFromArray<float>(TensorShape({}), {0.5});
75 AddInputFromArray<float>(TensorShape({}), {2});
76 AddInputFromArray<float>(TensorShape({}), {0.3});
80 Tensor expected(allocator(), DT_FLOAT, TensorShape({5}));
89 AddInputFromArray<double>(TensorShape({}), {0.0});
90 AddInputFromArray<double>(TensorShape({}), {10000})
[all...] |
| ragged_range_op_test.cc | 50 AddInputFromArray<int>(TensorShape({4}), {0, 5, 8, 5}); // starts
51 AddInputFromArray<int>(TensorShape({4}), {8, 7, 8, 1}); // limits
52 AddInputFromArray<int>(TensorShape({4}), {2, 1, 1, -1}); // deltas
65 AddInputFromArray<float>(TensorShape({4}), {0, 5, 8, 5}); // starts
66 AddInputFromArray<float>(TensorShape({4}), {8, 7, 8, 1}); // limits
67 AddInputFromArray<float>(TensorShape({4}), {2, 1, 1, -1}); // deltas
80 AddInputFromArray<int>(TensorShape({3}), {0, 5, 8}); // starts
81 AddInputFromArray<int>(TensorShape({3}), {8, 7, 8}); // limits
82 AddInputFromArray<int>(TensorShape({}), {1}); // deltas
95 AddInputFromArray<int>(TensorShape({}), {0}); // start
[all...] |
| batch_norm_op_test.cc | 46 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}),
48 AddInputFromArray<float>(TensorShape({2}), {10, 20});
49 AddInputFromArray<float>(TensorShape({2}), {0.25f, 0.5f});
50 AddInputFromArray<float>(TensorShape({2}), {0.1f, 0.6f});
51 AddInputFromArray<float>(TensorShape({2}), {0.0f, 0.0f});
54 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 6, 2}));
73 AddInputFromList<Eigen::half>(TensorShape({1, 1, 6, 2}),
75 AddInputFromList<Eigen::half>(TensorShape({2}), {10, 20});
76 AddInputFromList<Eigen::half>(TensorShape({2}), {0.25, 0.5});
77 AddInputFromList<Eigen::half>(TensorShape({2}), {0.1, 0.6})
[all...] |
| cross_op_test.cc | 43 AddInputFromArray<float>(TensorShape({3}), {0, 0, 0});
44 AddInputFromArray<float>(TensorShape({3}), {0, 0, 0});
47 Tensor expected(allocator(), DT_FLOAT, TensorShape({3}));
53 AddInputFromArray<float>(TensorShape({2, 3}), {1, 0, 0, /**/ 0, 1, 0});
54 AddInputFromArray<float>(TensorShape({2, 3}), {0, 1, 0, /**/ 1, 0, 0});
57 Tensor expected(allocator(), DT_FLOAT, TensorShape({2, 3}));
69 AddInputFromArray<float>(TensorShape({3}), {u1, u2, u3});
70 AddInputFromArray<float>(TensorShape({3}), {v1, v2, v3});
73 Tensor expected(allocator(), DT_FLOAT, TensorShape({3}));
90 AddInputFromArray<int>(TensorShape({2, 3}), {2, 0, 0, /**/ 0, 2, 0})
[all...] |
| quantize_op_test.cc | 40 AddInputFromArray<float>(TensorShape({7}),
43 AddInputFromArray<float>(TensorShape({1}), {0});
45 AddInputFromArray<float>(TensorShape({1}), {255.0f});
47 Tensor expected(allocator(), DT_QUINT8, TensorShape({7}));
63 AddInputFromArray<float>(TensorShape({8}),
65 AddInputFromArray<float>(TensorShape({1}), {-255.0f});
66 AddInputFromArray<float>(TensorShape({1}), {127.0f});
68 Tensor expected(allocator(), DT_QUINT8, TensorShape({8}));
77 Tensor expected_output_min(allocator(), DT_FLOAT, TensorShape({}));
81 Tensor expected_output_max(allocator(), DT_FLOAT, TensorShape({}));
[all...] |
| scatter_nd_op_test.cc | 54 // AddInputFromArray<string>(TensorShape({1}), {"Brain"});
55 // AddInputFromArray<int32>(TensorShape({1}), {0});
56 // AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"});
60 // Tensor expected(allocator(), DT_STRING, TensorShape({1}));
67 // AddInputFromArray<bool>(TensorShape({1}), {false});
68 // AddInputFromArray<int32>(TensorShape({1}), {0});
69 // AddInputFromArray<bool>(TensorShape({1}), {true});
73 // Tensor expected(allocator(), DT_BOOL, TensorShape({1}));
82 AddInputFromArray<float>(TensorShape({5, 3}),
84 AddInputFromArray<int32>(TensorShape({3, 1}), {0, 4, 2})
[all...] |
| gather_op_test.cc | 57 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4});
58 AddInputFromArray<int32>(TensorShape({}), {3});
59 AddInputFromArray<int32>(TensorShape({}), {0});
63 Tensor expected(allocator(), DT_FLOAT, TensorShape({}));
73 TensorShape({5}), {std::complex<float>(0, 10), std::complex<float>(1, 11),
76 AddInputFromArray<int32>(TensorShape({}), {3});
77 AddInputFromArray<int32>(TensorShape({}), {0});
81 Tensor expected(allocator(), DT_COMPLEX64, TensorShape({}));
91 AddInputFromArray<float>(TensorShape({5, 3}),
93 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2})
[all...] |
| identity_n_op_test.cc | 41 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
42 AddInputFromArray<double>(TensorShape({6}),
45 Tensor expected0(allocator(), DT_INT32, TensorShape({6}));
48 Tensor expected1(allocator(), DT_DOUBLE, TensorShape({6}));
55 AddInputFromArray<int32>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6});
56 AddInputFromArray<int32>(TensorShape({2, 3}), {7, 8, 9, 10, 11, 12});
58 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
67 AddInputFromArray<string>(TensorShape({6}), {"A", "b", "C", "d", "E", "f"});
68 AddInputFromArray<int32>(TensorShape({8}), {1, 3, 5, 7, 9, 11, 13, 15});
70 Tensor expected0(allocator(), DT_STRING, TensorShape({6}))
[all...] |
| scatter_op_test.cc | 53 AddInputFromArray<string>(TensorShape({1}), {"Brain"});
54 AddInputFromArray<int32>(TensorShape({1}), {0});
55 AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"});
59 Tensor expected(allocator(), DT_STRING, TensorShape({1}));
66 AddInputFromArray<bool>(TensorShape({1}), {false});
67 AddInputFromArray<int32>(TensorShape({1}), {0});
68 AddInputFromArray<bool>(TensorShape({1}), {true});
72 Tensor expected(allocator(), DT_BOOL, TensorShape({1}));
81 AddInputFromArray<float>(TensorShape({5, 3}),
83 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2})
[all...] |
| logging_ops_test.cc | 45 AddInputFromArray<string>(TensorShape({}), {"bar"});
70 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
71 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
72 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
74 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
81 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
82 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
83 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
85 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
92 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6})
[all...] |
| save_v2_op_test.cc | 62 AddInput<string>(TensorShape({}),
66 AddInput<string>(TensorShape({13}),
70 AddInput<string>(TensorShape({13}),
74 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; });
77 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; });
80 AddInput<float>(TensorShape({2, 4}),
84 AddInput<double>(TensorShape({2, 4}),
88 AddInput<qint8>(TensorShape({3, 2}),
92 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 {
97 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; })
[all...] |
| adjust_contrast_op_test.cc | 40 AddInputFromArray<float>(TensorShape({1, 1, 1, 3}), {-1, 2, 3});
41 AddInputFromArray<float>(TensorShape({}), {1.0});
44 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 3}));
55 AddInputFromArray<float>(TensorShape({1, 2, 2, 3}),
57 AddInputFromArray<float>(TensorShape({}), {0.2f});
60 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 2, 2, 3}));
79 AddInputFromArray<float>(TensorShape({1, 99, 99, 3}), values);
80 AddInputFromArray<float>(TensorShape({}), {0.2f});
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops_util.h | 27 Status MergeAxes(const TensorShape& broadcast_shape,
28 const TensorShape& storage_shape,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| shape_util.h | 25 // Converts a TensorShape to a constant Tensor.
27 // The input TensorShape input_shape is used to populate the elements of
29 Status TensorShapeToConstant(const TensorShape& input_shape,
|
| gather_op_helpers.h | 36 Status XlaGather(const xla::XlaOp& input, const TensorShape& input_shape,
37 const xla::XlaOp& indices, const TensorShape& indices_shape,
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape_test.cc | 29 static void set_data_type(TensorShape* s, DataType t) { s->set_data_type(t); }
30 static uint8 data_type(const TensorShape* s) { return s->data_type(); }
36 // The default TensorShape constructor constructs a shape of 0-dim
38 TensorShape s;
44 TensorShape s({10, 5});
58 TensorShape s({10, 5});
65 TensorShape s({10, 5, 20});
74 TensorShape s({2, 3, 5, 7});
86 TensorShape s0({2, 3, 5, 7});
101 TensorShape s1({2, 3, 5, 7})
[all...] |
| unique_tensor_references_test.cc | 25 Tensor a(DT_FLOAT, TensorShape({2, 2}));
26 Tensor b(DT_FLOAT, TensorShape({2, 2}));
48 Tensor a(DT_FLOAT, TensorShape({2, 2}));
67 Tensor a(DT_FLOAT, TensorShape({2, 2}));
68 Tensor b(DT_FLOAT, TensorShape({2, 2}));
78 Tensor a(DT_FLOAT, TensorShape({2, 2}));
79 Tensor b(DT_FLOAT, TensorShape({2, 2}));
80 Tensor c(DT_FLOAT, TensorShape({2, 2}));
81 Tensor d(DT_FLOAT, TensorShape({2, 2}));
82 Tensor e(DT_FLOAT, TensorShape({2, 2}))
[all...] |
