| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sets_test.py | 173 expected_indices = [[0, 0], [0, 1]]
183 expected_indices,
194 expected_indices,
209 expected_indices = [[0, 0], [0, 1]]
219 expected_indices,
233 expected_indices = [[0, 0]]
243 expected_indices,
254 expected_indices,
265 expected_indices,
367 expected_indices =
[all...] |
| topk_op_test.py | 45 expected_indices,
48 np_expected_indices = np.array(expected_indices)
|
| barrier_ops_test.py | 360 expected_indices = sorted(
363 self.assertAllEqual(all_indices, expected_indices)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_add_op_test.cc | 81 Tensor expected_indices(allocator(), DT_INT64, indices_shape);
82 test::FillValues<int64>(&expected_indices, indices);
83 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0));
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)); \
|
| sparse_reduce_sum_op_test.cc | 109 Tensor expected_indices(allocator(), DT_INT64,
111 test::FillValues<int64>(&expected_indices, {0, 1});
112 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0));
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor_test.cc | 463 std::vector<Eigen::Tensor<int64, 2, Eigen::RowMajor> > expected_indices; local
467 expected_indices.emplace_back(3, NDIM); // 3 x 3 tensor
469 expected_indices[0].setZero();
470 expected_indices[0](1, 2) = 2; // 002
471 expected_indices[0](2, 1) = 1; // 010
478 expected_indices.emplace_back(1, NDIM);
480 expected_indices[1].setZero();
481 expected_indices[1](0, 0) = 2; // 200
485 expected_indices.emplace_back(1, NDIM);
487 expected_indices[2].setZero()
497 << expected_indices[gix]; local
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| batch_dataset_op_test.py | 153 expected_indices = []
157 expected_indices.append([j, k])
160 indices=expected_indices,
|
| dataset_constructor_op_test.py | 317 expected_indices = np.array(
319 self.assertAllEqual(expected_indices, results.indices)
|
| /frameworks/base/libs/androidfw/tests/ |
| AttributeResolution_test.cpp | 271 std::array<uint32_t, 7> expected_indices = {{6u, 0u, 1u, 2u, 3u, 4u, 5u}}; local
272 EXPECT_EQ(expected_indices, indices);
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| sequence_queueing_state_saver_test.py | 622 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/xla/ |
| literal_util_test.cc | 208 Array2D<int64> expected_indices = { local
218 expected_indices.data(), expected_indices.num_elements()));
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers_test.py | [all...] |
