Home
last modified time | relevance | path

Searched refs:TensorShape (Results 1 – 25 of 1140) sorted by relevance

12345678910>>...46

/external/tensorflow/tensorflow/core/kernels/
Dnon_max_suppression_op_test.cc48 TensorShape({6, 4}), in TEST_F()
51 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
52 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
55 Tensor expected(allocator(), DT_INT32, TensorShape({3})); in TEST_F()
62 AddInputFromArray<float>(TensorShape({6, 4}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
66 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
69 Tensor expected(allocator(), DT_INT32, TensorShape({3})); in TEST_F()
77 TensorShape({6, 4}), in TEST_F()
80 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
[all …]
Dcrop_and_resize_op_test.cc53 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 …]
Dfake_quant_ops_test.cc29 using tensorflow::TensorShape;
35 void AddRandomInput(const TensorShape& shape) { in AddRandomInput()
54 const float max, const TensorShape& shape, in RunTestFakeQuantWithMinMaxArgs()
79 const float max, const TensorShape& shape, in RunTestFakeQuantWithMinMaxVars()
93 AddInputFromArray<float>(TensorShape({}), {min}); in RunTestFakeQuantWithMinMaxVars()
95 AddInputFromArray<float>(TensorShape({}), {max}); in RunTestFakeQuantWithMinMaxVars()
101 Tensor expected(allocator(), DT_FLOAT, TensorShape({2, 3})); in RunTestFakeQuantWithMinMaxVars()
108 const TensorShape& minmax_shape, const gtl::ArraySlice<float>& min, in RunTestFakeQuantWithMinMaxVarsPerChannel()
109 const gtl::ArraySlice<float>& max, const TensorShape& shape, in RunTestFakeQuantWithMinMaxVarsPerChannel()
142 8, false, -10.0f, 53.75f, TensorShape({2, 3}), in TEST_F()
[all …]
Droll_op_test.cc56 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F()
57 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
58 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
62 Tensor expected(allocator(), DT_FLOAT, TensorShape({5})); in TEST_F()
71 AddInputFromArray<string>(TensorShape({5}), {"a", "b", "c", "d", "e"}); in TEST_F()
72 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
73 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
77 Tensor expected(allocator(), DT_STRING, TensorShape({5})); in TEST_F()
87 TensorShape({5}), {std::complex<float>(0, 10), std::complex<float>(1, 11), in TEST_F()
90 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
[all …]
Dresize_nearest_neighbor_op_test.cc77 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
78 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
81 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1})); in TEST_F()
95 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
96 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
99 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1})); in TEST_F()
112 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
113 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F()
116 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 3, 3, 1})); in TEST_F()
133 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
[all …]
Dquantize_and_dequantize_op_test.cc52 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
53 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
54 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
57 Tensor expected(allocator(), DT_FLOAT, TensorShape({1})); in TEST_F()
77 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
78 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
79 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
80 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits in TEST_F()
83 Tensor expected(allocator(), DT_FLOAT, TensorShape({1})); in TEST_F()
104 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
[all …]
Dragged_range_op_test.cc50 AddInputFromArray<int>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
51 AddInputFromArray<int>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
52 AddInputFromArray<int>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
65 AddInputFromArray<float>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
66 AddInputFromArray<float>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
67 AddInputFromArray<float>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
80 AddInputFromArray<int>(TensorShape({3}), {0, 5, 8}); // starts in TEST_F()
81 AddInputFromArray<int>(TensorShape({3}), {8, 7, 8}); // limits in TEST_F()
82 AddInputFromArray<int>(TensorShape({}), {1}); // deltas in TEST_F()
95 AddInputFromArray<int>(TensorShape({}), {0}); // starts in TEST_F()
[all …]
Ddynamic_stitch_op_test.cc52 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F()
53 AddInputFromArray<int32>(TensorShape({5}), {1, 6, 2, 3, 5}); in TEST_F()
54 AddInputFromArray<float>(TensorShape({3}), {0, 40, 70}); in TEST_F()
55 AddInputFromArray<float>(TensorShape({5}), {10, 60, 20, 30, 50}); in TEST_F()
59 Tensor expected(allocator(), DT_FLOAT, TensorShape({8})); in TEST_F()
68 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F()
69 AddInputFromArray<int32>(TensorShape({2}), {1, 6}); in TEST_F()
70 AddInputFromArray<int32>(TensorShape({3}), {2, 3, 5}); in TEST_F()
71 AddInputFromArray<float>(TensorShape({3, 2}), {0, 1, 40, 41, 70, 71}); in TEST_F()
72 AddInputFromArray<float>(TensorShape({2, 2}), {10, 11, 60, 61}); in TEST_F()
[all …]
Dquantize_op_test.cc40 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
43 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
45 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
47 Tensor expected(allocator(), DT_QUINT8, TensorShape({7})); in TEST_F()
63 AddInputFromArray<float>(TensorShape({8}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({1}), {-255.0f}); in TEST_F()
66 AddInputFromArray<float>(TensorShape({1}), {127.0f}); in TEST_F()
68 Tensor expected(allocator(), DT_QUINT8, TensorShape({8})); in TEST_F()
77 Tensor expected_output_min(allocator(), DT_FLOAT, TensorShape({})); in TEST_F()
81 Tensor expected_output_max(allocator(), DT_FLOAT, TensorShape({})); in TEST_F()
[all …]
Dgather_op_test.cc57 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F()
58 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
59 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
63 Tensor expected(allocator(), DT_FLOAT, TensorShape({})); in TEST_F()
73 TensorShape({5}), {std::complex<float>(0, 10), std::complex<float>(1, 11), in TEST_F()
76 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
77 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
81 Tensor expected(allocator(), DT_COMPLEX64, TensorShape({})); in TEST_F()
91 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
93 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2}); in TEST_F()
[all …]
Dsave_v2_op_test.cc62 AddInput<string>(TensorShape({}), in TEST_F()
66 AddInput<string>(TensorShape({13}), in TEST_F()
70 AddInput<string>(TensorShape({13}), in TEST_F()
74 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; }); in TEST_F()
77 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; }); in TEST_F()
80 AddInput<float>(TensorShape({2, 4}), in TEST_F()
84 AddInput<double>(TensorShape({2, 4}), in TEST_F()
88 AddInput<qint8>(TensorShape({3, 2}), in TEST_F()
92 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 { in TEST_F()
97 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; }); in TEST_F()
[all …]
Dsequence_ops_test.cc59 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
60 AddInputFromArray<int32>(TensorShape({}), {10}); in TEST_F()
61 AddInputFromArray<int32>(TensorShape({}), {2}); in TEST_F()
65 Tensor expected(allocator(), DT_INT32, TensorShape({5})); in TEST_F()
74 AddInputFromArray<float>(TensorShape({}), {0.5}); in TEST_F()
75 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F()
76 AddInputFromArray<float>(TensorShape({}), {0.3}); in TEST_F()
80 Tensor expected(allocator(), DT_FLOAT, TensorShape({5})); in TEST_F()
89 AddInputFromArray<double>(TensorShape({}), {0.0}); in TEST_F()
90 AddInputFromArray<double>(TensorShape({}), {10000}); in TEST_F()
[all …]
Dresize_bilinear_op_test.cc46 const Tensor* SetRandomImageInput(const TensorShape& shape) { in SetRandomImageInput()
124 const TensorShape shape({batch_size, input_width, input_height, channels}); in TestResize()
126 AddInputFromArray<int32>(TensorShape({2}), {output_width, output_height}); in TestResize()
132 TensorShape({batch_size, output_width, output_height, channels}))); in TestResize()
193 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
194 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
199 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 1, 1})); in TEST_F()
205 const Tensor* input = SetRandomImageInput(TensorShape({1, 2, 2, 1})); in TEST_F()
206 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
214 DataTypeToEnum<float>::v(), TensorShape({1, 1, 1, 1}))); in TEST_F()
[all …]
Dscatter_op_test.cc53 AddInputFromArray<string>(TensorShape({1}), {"Brain"}); in TEST_F()
54 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
55 AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"}); in TEST_F()
59 Tensor expected(allocator(), DT_STRING, TensorShape({1})); in TEST_F()
66 AddInputFromArray<bool>(TensorShape({1}), {false}); in TEST_F()
67 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
68 AddInputFromArray<bool>(TensorShape({1}), {true}); in TEST_F()
72 Tensor expected(allocator(), DT_BOOL, TensorShape({1})); in TEST_F()
81 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
83 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2}); in TEST_F()
[all …]
Dsave_op_test.cc68 AddInput<string>(TensorShape({}), in TEST_F()
72 AddInput<string>(TensorShape({14}), in TEST_F()
76 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; }); in TEST_F()
79 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; }); in TEST_F()
82 AddInput<float>(TensorShape({2, 4}), in TEST_F()
86 AddInput<double>(TensorShape({2, 4}), in TEST_F()
90 AddInput<qint8>(TensorShape({3, 2}), in TEST_F()
94 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 { in TEST_F()
99 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; }); in TEST_F()
102 AddInput<int8>(TensorShape({7}), [](int x) -> int8 { return x - 7; }); in TEST_F()
[all …]
Dscatter_nd_op_test.cc82 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
84 AddInputFromArray<int32>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F()
85 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
91 Tensor expected(allocator(), DT_FLOAT, TensorShape({5, 3})); in TEST_F()
101 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
103 AddInputFromArray<int64>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F()
104 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
110 Tensor expected(allocator(), DT_FLOAT, TensorShape({5, 3})); in TEST_F()
133 AddInputFromArray<float>(TensorShape({5}), {0, 0, 0, 0, 0}); in TEST_F()
134 AddInputFromArray<int32>(TensorShape({1}), {3}); in TEST_F()
[all …]
Dfused_batch_norm_op_test.cc43 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
45 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
46 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F()
47 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
48 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
52 Tensor expected(allocator(), DT_FLOAT, TensorShape({1, 1, 6, 2})); in TEST_F()
57 Tensor expected_mean(allocator(), DT_FLOAT, TensorShape({2})); in TEST_F()
61 Tensor expected_variance(allocator(), DT_FLOAT, TensorShape({2})); in TEST_F()
77 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
79 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
[all …]
Dquantized_matmul_op_test.cc58 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6}); in TEST_F()
63 AddInputFromArray<quint8>(TensorShape({3, 4}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
66 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
67 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
68 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
80 Tensor expected(allocator(), DT_QINT32, TensorShape({2, 4})); in TEST_F()
116 AddInputFromArray<quint8>(TensorShape({a_rows, a_cols}), {11}); in TEST_F()
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F()
121 AddInputFromArray<float>(TensorShape({1}), {-12.0f}); in TEST_F()
[all …]
Dsparse_dense_binary_op_shared_test.cc75 AddInputFromArray<int64>(TensorShape({1, 1}), {0}); // indices in TEST_F()
76 AddInputFromArray<float>(TensorShape({1}), {1618}); // values in TEST_F()
77 AddInputFromArray<int64>(TensorShape({1}), {1}); // shape in TEST_F()
78 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); // dense in TEST_F()
86 AddInputFromArray<int64>(TensorShape({1, 2}), {0, 0}); in TEST_F()
87 AddInputFromArray<float>(TensorShape({1}), {1618}); in TEST_F()
88 AddInputFromArray<int64>(TensorShape({2}), {1, 1}); in TEST_F()
89 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); in TEST_F()
99 const auto indices_shape = TensorShape({4, 2}); in TEST_F()
106 Tensor dense(DT_FLOAT, TensorShape(shape)); in TEST_F()
[all …]
Dragged_gather_op_test.cc33 const TensorShape& indices_shape, const std::vector<INDEX_TYPE>& indices, in BuildRaggedGatherGraph()
35 const TensorShape& params_dense_values_shape, in BuildRaggedGatherGraph()
54 AddInputFromArray<int64>(TensorShape({splits_size}), splits); in BuildRaggedGatherGraph()
67 TensorShape({4}), // indices.shape in TEST_F()
70 TensorShape({9}), // params_dense_values.shape in TEST_F()
89 TensorShape({5}), // indices.shape in TEST_F()
92 TensorShape({9}), // params_dense_values.shape in TEST_F()
113 TensorShape({4}), // indices.shape in TEST_F()
116 TensorShape({4, 2}), // params_dense_values.shape in TEST_F()
132 test::AsTensor<int32>({1, 2, 3, 4, 5, 6, 7, 8}, TensorShape({4, 2}))); in TEST_F()
[all …]
Drequantize_op_test.cc54 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F()
56 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F()
57 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F()
58 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F()
59 AddInputFromArray<float>(TensorShape({1}), {1.0f}); in TEST_F()
61 Tensor expected(allocator(), DT_QUINT8, TensorShape({value_count})); in TEST_F()
72 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F()
74 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F()
75 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F()
76 AddInputFromArray<float>(TensorShape({1}), {0.01f}); in TEST_F()
[all …]
Dlogging_ops_test.cc45 AddInputFromArray<string>(TensorShape({}), {"bar"}); in TEST_F()
70 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
71 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
72 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
74 Tensor expected(allocator(), DT_INT32, TensorShape({6})); in TEST_F()
81 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
82 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
83 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
85 Tensor expected(allocator(), DT_INT32, TensorShape({6})); in TEST_F()
92 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc29 static void set_data_type(TensorShape* s, DataType t) { s->set_data_type(t); } in set_data_type()
30 static uint8 data_type(const TensorShape* s) { return s->data_type(); } in data_type()
38 TensorShape s; in TEST()
44 TensorShape s({10, 5}); in TEST()
58 TensorShape s({10, 5}); in TEST()
65 TensorShape s({10, 5, 20}); in TEST()
74 TensorShape s({2, 3, 5, 7}); in TEST()
86 TensorShape s0({2, 3, 5, 7}); in TEST()
101 TensorShape s1({2, 3, 5, 7}); in TEST()
107 TensorShape s2({2, 3, 5, 7}); in TEST()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dmath_grad_test.cc417 TensorShape({2, 3})); in TEST_F()
420 {g(-3.f), g(-2.f), g(-1.f), g(1.f), g(2.f), g(3.f)}, TensorShape({2, 3})); in TEST_F()
427 TensorShape({2, 3})); in TEST_F()
430 {g(-3.f), g(-2.f), g(-1.f), g(1.f), g(2.f), g(3.f)}, TensorShape({2, 3})); in TEST_F()
437 TensorShape({2, 3})); in TEST_F()
440 {g(-3.f), g(-2.f), g(-1.f), g(1.f), g(2.f), g(3.f)}, TensorShape({2, 3})); in TEST_F()
447 TensorShape({2, 3})); in TEST_F()
450 {g(-3.f), g(-2.f), g(-1.f), g(1.f), g(2.f), g(3.f)}, TensorShape({2, 3})); in TEST_F()
457 TensorShape({2, 3})); in TEST_F()
460 {g(1.f), g(2.f), g(3.f), g(4.f), g(5.f), g(6.f)}, TensorShape({2, 3})); in TEST_F()
[all …]
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad_test.cc35 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
36 const TensorShape& y_shape) { in RunTest()
44 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
45 const OutputList& ys, const std::vector<TensorShape>& y_shapes) { in RunTest()
57 TensorShape x_shape({1, 2, 3}); in TEST_F()
62 TensorShape y_shape({2, 1, 2, 3}); in TEST_F()
67 TensorShape x_shape({1, 2, 3}); in TEST_F()
72 TensorShape y_shape({1, 2, 2, 3}); in TEST_F()
77 TensorShape x_shape({4, 2, 3}); in TEST_F()
80 std::vector<TensorShape> y_shapes(4, TensorShape({2, 3})); in TEST_F()
[all …]

12345678910>>...46