Home
last modified time | relevance | path

Searched refs:expected_indices (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsets_test.py177 expected_indices = [[0, 0], [0, 1]]
187 expected_indices,
198 expected_indices,
214 expected_indices = [[0, 0], [0, 1]]
224 expected_indices,
239 expected_indices = [[0, 0]]
249 expected_indices,
260 expected_indices,
271 expected_indices,
374 expected_indices = [
[all …]
Dtopk_op_test.py46 expected_indices, argument
49 np_expected_indices = np.array(expected_indices)
Dbarrier_ops_test.py372 expected_indices = sorted(
375 self.assertAllEqual(all_indices, expected_indices)
/external/tensorflow/tensorflow/core/kernels/
Dsparse_add_op_test.cc82 Tensor expected_indices(allocator(), DT_INT64, indices_shape); in TEST_F() local
83 test::FillValues<int64>(&expected_indices, indices); in TEST_F()
84 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); in TEST_F()
129 Tensor expected_indices(allocator(), DT_INT64, \
131 test::FillValues<int64>(&expected_indices, \
133 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \
189 Tensor expected_indices(allocator(), DT_INT64, TensorShape({0, 2})); \
190 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \
Dsparse_reduce_sum_op_test.cc110 Tensor expected_indices(allocator(), DT_INT64, in TEST_F() local
112 test::FillValues<int64>(&expected_indices, {0, 1}); in TEST_F()
113 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); in TEST_F()
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dtable_utils_test.py95 expected_indices = [[0, 0], [1, 2]]
103 self.assertAllEqual(expected_indices, output_data.indices)
114 expected_indices = [[0, 0], [1, 2]]
122 self.assertAllEqual(expected_indices, output_data.indices)
170 expected_indices = [[0, 0], [1, 2]]
178 self.assertAllEqual(expected_indices, output_data.indices)
189 expected_indices = [[0, 0], [1, 2]]
197 self.assertAllEqual(expected_indices, output_data.indices)
346 expected_indices = [[0, 0], [1, 2]]
357 self.assertAllEqual(expected_indices, output_data.indices)
[all …]
Dcategory_encoding_test.py57 expected_indices = [[0, 1], [0, 2], [0, 3], [1, 0], [1, 3]]
69 self.assertAllEqual(expected_indices, sp_output_dataset.indices)
143 expected_indices = [[0, 0], [1, 2], [2, 1], [3, 0]]
154 self.assertAllEqual(expected_indices, sp_output_dataset.indices)
182 expected_indices = [[0, 0], [1, 2], [2, 1], [3, 0]]
193 self.assertAllEqual(expected_indices, sp_output_dataset.indices)
223 expected_indices = [[0, 1], [0, 2], [0, 3], [1, 3]]
235 self.assertAllEqual(expected_indices, sp_output_dataset.indices)
Dinteger_lookup_test.py147 expected_indices = [[0, 0], [1, 2]]
157 self.assertAllEqual(expected_indices, output_data.indices)
188 expected_indices = [[0, 0], [1, 2]]
203 self.assertAllEqual(expected_indices, output_data.indices)
255 expected_indices = [[0, 0], [1, 2]]
265 self.assertAllEqual(expected_indices, output_data.indices)
Dindex_lookup_test.py384 expected_indices = [[0, 0], [1, 2]]
399 self.assertAllEqual(expected_indices, output_data.indices)
410 expected_indices = [[0, 0], [1, 2]]
425 self.assertAllEqual(expected_indices, output_data.indices)
499 expected_indices = [[0, 0], [1, 2]]
514 self.assertAllEqual(expected_indices, output_data.indices)
525 expected_indices = [[0, 0], [1, 2]]
540 self.assertAllEqual(expected_indices, output_data.indices)
627 expected_indices = [[0, 0], [1, 2]]
642 self.assertAllEqual(expected_indices, output_data.indices)
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor_test.cc493 std::vector<Eigen::Tensor<int64, 2, Eigen::RowMajor> > expected_indices; in TEST() local
497 expected_indices.emplace_back(3, NDIM); // 3 x 3 tensor in TEST()
499 expected_indices[0].setZero(); in TEST()
500 expected_indices[0](1, 2) = 2; // 002 in TEST()
501 expected_indices[0](2, 1) = 1; // 010 in TEST()
508 expected_indices.emplace_back(1, NDIM); in TEST()
510 expected_indices[1].setZero(); in TEST()
511 expected_indices[1](0, 0) = 2; // 200 in TEST()
515 expected_indices.emplace_back(1, NDIM); in TEST()
517 expected_indices[2].setZero(); in TEST()
[all …]
/external/fonttools/Tests/ttLib/tables/
DT_S_I__0_test.py64 def test_decompile(table, numGlyphs, data, expected_indices, argument
71 assert table.indices == expected_indices
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_sparse_tensor_slices_test.py61 expected_indices = np.array(
63 self.assertAllEqual(expected_indices, results.indices)
Dbatch_test.py147 expected_indices = []
151 expected_indices.append([j, k])
155 indices=expected_indices,
Dwindow_test.py160 expected_indices = []
164 expected_indices.append([j, k])
168 indices=expected_indices,
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op_gpu_test.cc465 const auto expected_indices = test::AsTensor<int>({3, 0, 5, 0, 0}); in TEST_F() local
466 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
484 const auto expected_indices = test::AsTensor<int>({3, 0, 0, 0, 0, 0}); in TEST_F() local
485 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
Dnon_max_suppression_op_test.cc613 const auto expected_indices = test::AsTensor<int>({3, 0, 5, 0, 0}); in TEST_F() local
614 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
631 const auto expected_indices = test::AsTensor<int>({3, 0, 0, 0, 0, 0}); in TEST_F() local
632 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
670 const auto expected_indices = test::AsTensor<int>({3, 0, 5, 0, 0}); in TEST_F() local
671 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
/external/tensorflow/tensorflow/python/ops/
Dbincount_ops_test.py151 expected_indices, argument
166 self.assertAllEqual(expected_indices, y.indices)
347 expected_indices, argument
364 self.assertAllEqual(expected_indices, y.indices)
494 expected_indices, argument
511 self.assertAllEqual(expected_indices, y.indices)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dbucket_by_sequence_length_test.py432 expected_indices = []
436 expected_indices.append((length - batch_start_len, val))
438 expected_sprs_tensor = (tuple(expected_indices),
/external/tensorflow/tensorflow/python/keras/utils/
Dcomposite_tensor_support_test.py260 expected_indices = np.array([[0, 0], [1, 0], [1, 1]])
264 self.assertAllEqual(output.indices, expected_indices)
279 expected_indices = np.array([[0, 0], [1, 0], [1, 1], [2, 0], [3, 0], [3,
284 self.assertAllEqual(output.indices, expected_indices)
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc493 const int32 expected_indices[] = {0, 2}; in TEST() local
496 EXPECT_EQ(expected_indices[0], fn.result1(0)); in TEST()
497 EXPECT_EQ(expected_indices[1], fn.result1(1)); in TEST()
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py583 expected_indices = (
589 self.assertAllEqual(expected_indices, packed_input_indices)
597 self.assertAllEqual(expected_indices, packed_output_indices)
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py2325 expected_indices = np.array([[0, 0], [1, 0], [1, 1]])
2329 self.assertAllEqual(output.indices, expected_indices)