Home
last modified time | relevance | path

Searched refs:expected_indices (Results 1 – 17 of 17) 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.py370 expected_indices = sorted(
373 self.assertAllEqual(all_indices, expected_indices)
/external/tensorflow/tensorflow/core/kernels/
Dsparse_add_op_test.cc81 Tensor expected_indices(allocator(), DT_INT64, indices_shape); in TEST_F() local
82 test::FillValues<int64>(&expected_indices, indices); in TEST_F()
83 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); in TEST_F()
128 Tensor expected_indices(allocator(), DT_INT64, \
130 test::FillValues<int64>(&expected_indices, \
132 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \
188 Tensor expected_indices(allocator(), DT_INT64, TensorShape({0, 2})); \
189 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \
Dsparse_reduce_sum_op_test.cc109 Tensor expected_indices(allocator(), DT_INT64, in TEST_F() local
111 test::FillValues<int64>(&expected_indices, {0, 1}); in TEST_F()
112 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); in TEST_F()
Dnon_max_suppression_op_test.cc603 const auto expected_indices = test::AsTensor<int>({3, 0, 5, 0, 0}); in TEST_F() local
604 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
621 const auto expected_indices = test::AsTensor<int>({3, 0, 0, 0, 0, 0}); in TEST_F() local
622 test::ExpectTensorEqual<int>(expected_indices, *GetOutput(0)); in TEST_F()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor_test.cc489 std::vector<Eigen::Tensor<int64, 2, Eigen::RowMajor> > expected_indices; in TEST() local
493 expected_indices.emplace_back(3, NDIM); // 3 x 3 tensor in TEST()
495 expected_indices[0].setZero(); in TEST()
496 expected_indices[0](1, 2) = 2; // 002 in TEST()
497 expected_indices[0](2, 1) = 1; // 010 in TEST()
504 expected_indices.emplace_back(1, NDIM); in TEST()
506 expected_indices[1].setZero(); in TEST()
507 expected_indices[1](0, 0) = 2; // 200 in TEST()
511 expected_indices.emplace_back(1, NDIM); in TEST()
513 expected_indices[2].setZero(); in TEST()
[all …]
/external/fonttools/Tests/ttLib/tables/
DT_S_I__0_test.py65 def test_decompile(table, numGlyphs, data, expected_indices, argument
72 assert table.indices == expected_indices
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dbatch_test.py126 expected_indices = []
130 expected_indices.append([j, k])
134 indices=expected_indices,
Dfrom_sparse_tensor_slices_test.py57 expected_indices = np.array(
59 self.assertAllEqual(expected_indices, results.indices)
Dwindow_test.py160 expected_indices = []
164 expected_indices.append([j, k])
168 indices=expected_indices,
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dslide_dataset_op_test.py249 expected_indices = []
253 expected_indices.append([j, k])
256 indices=expected_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/contrib/training/python/training/
Dsequence_queueing_state_saver_test.py622 expected_indices = [[], [0], [0, 1], [1, 2, 3], [2, 3, 4, 5],
625 self.assertEqual(len(all_received_indices), len(expected_indices))
626 for received, expected in zip(all_received_indices, expected_indices):
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc471 const int32 expected_indices[] = {0, 2}; in TEST() local
474 EXPECT_EQ(expected_indices[0], fn.result1(0)); in TEST()
475 EXPECT_EQ(expected_indices[1], fn.result1(1)); in TEST()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc268 Array2D<int64> expected_indices = { in TEST_F() local
277 absl::Span<const int64>(expected_indices.data(), in TEST_F()
278 expected_indices.num_elements())); in TEST_F()
286 absl::Span<const int64>(expected_indices.data(), in TEST_F()
287 expected_indices.num_elements())); in TEST_F()
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py1570 expected_indices, expected_values, _ = _sparsify(reshaped_random_)
1579 np.testing.assert_array_equal(expected_indices, flattened.indices)