| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_test.py | 126 expected_outputs, argument
157 self.assertEqual(actual_outputs, expected_outputs)
164 expected_outputs = [1, 2, 2, 4, 4, 8]
165 self._test_helper(inputs, expected_outputs)
176 expected_outputs = [
181 self._test_helper(inputs, expected_outputs, init_loss_scale)
189 expected_outputs = [512, 256, 128, 64, 32, 16]
191 self._test_helper(inputs, expected_outputs, init_loss_scale)
199 expected_outputs = [8, 4, 2, 1, 1, 1]
201 self._test_helper(inputs, expected_outputs, init_loss_scale)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | repeat_dataset_op_test.cc | 137 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
153 if (expected_outputs_it == test_case.expected_outputs.end()) { in TEST_P()
154 expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
165 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
171 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
315 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
336 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
343 expected_outputs_it == test_case.expected_outputs.end()) { in TEST_P()
344 expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
353 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
|
| D | dataset_test_base.h | 373 GetNextTestCase(T dataset_params, std::vector<Tensor> expected_outputs,
376 expected_outputs(std::move(expected_outputs)), in dataset_params()
380 std::vector<Tensor> expected_outputs; member
387 bool get_next = false, std::vector<Tensor> expected_outputs = {},
393 expected_outputs(std::move(expected_outputs)), in dataset_params()
400 std::vector<Tensor> expected_outputs; member
460 std::vector<Tensor> expected_outputs,
464 expected_outputs(std::move(expected_outputs)), in dataset_params()
469 std::vector<Tensor> expected_outputs; member
572 Status CheckIteratorGetNext(const std::vector<Tensor>& expected_outputs,
[all …]
|
| D | sparse_tensor_slice_dataset_op_test.cc | 108 std::vector<std::vector<Tensor>> expected_outputs; member
170 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
181 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
307 std::vector<std::vector<Tensor>> expected_outputs; member
391 out_tensors[0], test_case.expected_outputs[cur_iteration - 1][0])); in TEST_P()
393 out_tensors[1], test_case.expected_outputs[cur_iteration - 1][1])); in TEST_P()
395 out_tensors[2], test_case.expected_outputs[cur_iteration - 1][2])); in TEST_P()
|
| D | window_dataset_op_test.cc | 237 std::vector<std::vector<Tensor>> expected_outputs; member
290 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
318 EXPECT_LT(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
324 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
405 std::vector<std::vector<Tensor>> expected_outputs; member
473 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
511 EXPECT_LT(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
521 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
|
| D | tensor_dataset_op_test.cc | 132 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
139 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>(); in TEST_P()
142 TF_EXPECT_OK(ExpectEqual(out_tensors[i], test_case.expected_outputs[i])); in TEST_P()
262 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
269 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>(); in TEST_P()
272 TF_EXPECT_OK(ExpectEqual(out_tensors[i], test_case.expected_outputs[i])); in TEST_P()
|
| D | dataset_test_base.cc | 577 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext() argument
579 expected_outputs, compare_order); in CheckIteratorGetNext()
583 TestIterator* iterator, const std::vector<Tensor>& expected_outputs, in CheckIteratorGetNext() argument
586 expected_outputs, compare_order); in CheckIteratorGetNext()
591 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext() argument
605 TF_EXPECT_OK(ExpectEqual(out_tensors, expected_outputs, in CheckIteratorGetNext()
612 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorSkip() argument
625 TF_EXPECT_OK(ExpectEqual(out_tensors, expected_outputs, in CheckIteratorSkip()
632 const DatasetParams& params, const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderFullIteration() argument
640 TF_RETURN_IF_ERROR(CheckIteratorGetNext(iterator.get(), expected_outputs, in CheckSplitProviderFullIteration()
[all …]
|
| D | cache_dataset_ops_test.cc | 180 TF_EXPECT_OK(ExpectEqual(out_tensors, test_case.expected_outputs, in TEST_P()
195 TF_EXPECT_OK(ExpectEqual(out_tensors, test_case.expected_outputs, in TEST_P()
332 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
348 EXPECT_LT(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
357 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
|
| D | tensor_slice_dataset_op_test.cc | 115 test_case.expected_outputs.size()); in TEST_P()
121 test_case.expected_outputs[i + num_tensors_per_slice * cur_slice] in TEST_P()
128 test_case.expected_outputs[i + num_tensors_per_slice * cur_slice])); in TEST_P()
292 .expected_outputs[i + in TEST_P()
300 test_case.expected_outputs[i + num_tensors_per_slice * in TEST_P()
|
| D | optimize_dataset_op_test.cc | 86 std::vector<Tensor> expected_outputs = in TEST_F() local
90 TF_EXPECT_OK(CheckIteratorGetNext(expected_outputs, /*compare_order=*/true)); in TEST_F()
|
| D | map_defun_op_test.cc | 118 std::vector<Tensor> expected_outputs; member
257 EXPECT_EQ(context->num_outputs(), test_case.expected_outputs.size()); in TEST_P()
260 test_case.expected_outputs[i])); in TEST_P()
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils_test.py | 81 expected_outputs = {model.output_names[0]: model(inputs)}
83 expected_outputs = {'output_1': model(inputs)}
85 self._assert_all_close(expected_outputs, signature_outputs)
106 expected_outputs = {model.output_names[0]: model(inputs)}
108 expected_outputs = {'output_1': model(inputs)}
110 self._assert_all_close(expected_outputs, signature_outputs)
150 expected_outputs = {
155 expected_outputs = {'output_1': outputs[0], 'output_2': outputs[1]}
156 self._assert_all_close(expected_outputs, signature_outputs)
191 expected_outputs = {model.output_names[0]: model(inputs)}
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | parse_single_example_op_test.py | 252 expected_outputs = [{
272 for proto, expected_output in zip(original, expected_outputs):
302 expected_outputs = [{
316 for proto, expected_output in zip(original, expected_outputs):
339 expected_outputs = [{
351 for proto, expected_output in zip(original, expected_outputs):
390 expected_outputs = [{
404 for proto, expected_output in zip(original, expected_outputs):
427 expected_outputs = [{
435 for proto, expected_output in zip(original, expected_outputs):
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_multistream_test.cc | 40 std::vector<int32> expected_outputs(n); in TEST() local
68 expected_outputs[i] = -inputs[i]; in TEST()
99 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST()
101 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone_test.cc | 274 std::vector<int64> expected_outputs; in TEST() member
296 test_case.expected_outputs[num_outputs]); in TEST()
298 EXPECT_EQ(test_case.expected_outputs.size(), num_outputs); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_init_op.cc | 58 DataTypeVector expected_outputs = {}; in Compute() local
59 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
126 DataTypeVector expected_outputs = {}; in Compute() local
127 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
|
| D | barrier_ops.cc | 593 DataTypeVector expected_outputs = {DT_INT64, DT_STRING}; in ComputeAsync() local
595 expected_outputs.push_back(dt); in ComputeAsync()
598 ctx, ctx->MatchSignature(expected_inputs, expected_outputs), callback); in ComputeAsync()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model_test.cc | 278 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
280 BuildSignatureDef(expected_inputs, expected_outputs); in TEST_F()
289 EXPECT_EQ(expected_outputs, outputs); in TEST_F()
308 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
310 EXPECT_EQ(expected_outputs, outputs); in TEST_F()
450 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0"}; in TEST_F() local
452 EXPECT_EQ(expected_outputs, outputs); in TEST_F()
480 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0", in TEST_F() local
483 EXPECT_EQ(expected_outputs, outputs); in TEST_F()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | cache_dataset_serialization_test.py | 54 def expected_outputs(self): member in CacheDatasetSerializationTest
75 self.assertSequenceEqual(outputs, self.expected_outputs())
95 self.assertSequenceEqual(outputs, self.expected_outputs())
115 self.assertSequenceEqual(outputs, self.expected_outputs())
209 self.assertSequenceEqual(outputs, self.expected_outputs())
|
| D | parallel_interleave_dataset_serialization_test.py | 70 expected_outputs = np.repeat(
78 self.assertSequenceEqual(sorted(actual), expected_outputs)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | spacetobatch_op_test.py | 169 expected_outputs = np.array(outputs).astype(dtype)
173 expected_outputs = outputs
179 sess.run(x_tf, {placeholder: actual_inputs}), expected_outputs)
185 sess.run(x_tf, {placeholder: expected_outputs}), actual_inputs)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | batch_test.py | 148 expected_outputs = []
152 expected_outputs.append(i * 5 + j)
156 values=expected_outputs,
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | random_dataset_op_test.cc | 129 while (out_tensors.size() < test_case.expected_outputs.size()) { in TEST_P()
139 TF_ASSERT_OK(ExpectEqual(out_tensors, test_case.expected_outputs, in TEST_P()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device.cc | 112 int expected_outputs = TFE_OpGetOutputLength(op.get(), "outputs", status); in ExecuteWithSpecialOps() local
114 if (expected_outputs != parallel_device.num_underlying_devices()) { in ExecuteWithSpecialOps()
118 " outputs for TPUReplicatedOutput, but got ", expected_outputs)); in ExecuteWithSpecialOps()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 64 const DataTypeSlice expected_outputs, in MatchSignatureHelper() argument
76 if (outputs.size() != expected_outputs.size()) signature_mismatch = true; in MatchSignatureHelper()
78 if (!TypesCompatible(expected_outputs[i], outputs[i])) { in MatchSignatureHelper()
88 DataTypeSliceString(expected_outputs)); in MatchSignatureHelper()
255 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature() argument
256 return MatchSignatureHelper(expected_inputs, expected_outputs, in MatchSignature()
1032 const DataTypeSlice expected_outputs) { in MatchSignature() argument
1038 return MatchSignatureHelper(expected_inputs, expected_outputs, inputs, in MatchSignature()
|