Home
last modified time | relevance | path

Searched refs:expected_outputs (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/mixed_precision/python/
Dloss_scale_manager_test.py60 expected_outputs, argument
85 self.assertEqual(actual_outputs, expected_outputs)
90 expected_outputs = [1, 2, 2, 4, 4, 8]
91 self._test_helper(inputs, expected_outputs)
100 expected_outputs = [
105 self._test_helper(inputs, expected_outputs, init_loss_scale)
111 expected_outputs = [1024, 512, 512, 256, 256, 128]
113 self._test_helper(inputs, expected_outputs, init_loss_scale)
119 expected_outputs = [16, 8, 8, 4, 4, 2, 2, 1, 1, 1]
121 self._test_helper(inputs, expected_outputs, init_loss_scale)
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc70 std::vector<SparseTensorParam> expected_outputs; member
134 std::vector<SparseTensorParam> expected_outputs = in TEST_F() local
135 test_case.expected_outputs; in TEST_F()
166 ExpectEqual(out_tensors[0], expected_outputs[cur_slice].indices)); in TEST_F()
168 ExpectEqual(out_tensors[1], expected_outputs[cur_slice].values)); in TEST_F()
170 expected_outputs[cur_slice].dense_shape)); in TEST_F()
209 std::vector<SparseTensorParam> expected_outputs = in TEST_F() local
210 test_case.expected_outputs; in TEST_F()
228 expected_outputs[0].indices.dtype(), expected_outputs[0].values.dtype(), in TEST_F()
229 expected_outputs[0].dense_shape.dtype()}; in TEST_F()
[all …]
Dtensor_slice_dataset_op_test.cc73 std::vector<Tensor> expected_outputs; member
134 std::vector<Tensor> expected_outputs = test_case.expected_outputs; in TEST_F() local
148 shapes.emplace_back(expected_outputs[i].shape()); in TEST_F()
179 expected_outputs.size()); in TEST_F()
186 expected_outputs[i + num_tensors_per_slice * cur_slice] in TEST_F()
193 expected_outputs[i + num_tensors_per_slice * cur_slice])); in TEST_F()
233 std::vector<Tensor> expected_outputs = test_case.expected_outputs; in TEST_F() local
247 shapes.emplace_back(expected_outputs[i].shape()); in TEST_F()
267 EXPECT_EQ(produced_output_dtypes[i], expected_outputs[i].dtype()); in TEST_F()
276 std::vector<Tensor> expected_outputs = test_case.expected_outputs; in TEST_F() local
[all …]
Dzip_dataset_op_test.cc84 std::vector<Tensor> expected_outputs; member
171 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
179 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
185 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
208 expected_output_dtypes.emplace_back(test_case.expected_outputs[i].dtype()); in TEST_P()
226 expected_output_shapes.emplace_back(test_case.expected_outputs[i].shape()); in TEST_P()
242 test_case.expected_outputs.size() / num_tensors_per_slice); in TEST_P()
270 expected_output_dtypes.emplace_back(test_case.expected_outputs[i].dtype()); in TEST_P()
288 expected_output_shapes.emplace_back(test_case.expected_outputs[i].shape()); in TEST_P()
308 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
[all …]
Dconcatenate_dataset_op_test.cc74 std::vector<Tensor> expected_outputs; member
186 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
194 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
200 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
308 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
331 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
341 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
Drepeat_dataset_op_test.cc67 std::vector<Tensor> expected_outputs; member
164 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
179 if (expected_outputs_it == test_case.expected_outputs.end()) { in TEST_P()
180 expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
191 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
197 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
478 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
528 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
535 expected_outputs_it == test_case.expected_outputs.end()) { in TEST_P()
536 expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
[all …]
Dtake_dataset_op_test.cc64 std::vector<Tensor> expected_outputs; member
190 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
198 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
204 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
486 auto expected_outputs_it = test_case.expected_outputs.begin(); in TEST_P()
536 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
544 if (breakpoint >= test_case.expected_outputs.size()) { in TEST_P()
546 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end()); in TEST_P()
Dtensor_dataset_op_test.cc71 std::vector<Tensor> expected_outputs; member
169 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
176 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>(); in TEST_P()
179 TF_EXPECT_OK(ExpectEqual(out_tensors[i], test_case.expected_outputs[i])); in TEST_P()
501 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
509 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>(); in TEST_P()
513 ExpectEqual(out_tensors[i], test_case.expected_outputs[i])); in TEST_P()
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py72 expected_outputs = {model.output_names[0]: model(inputs)}
74 self._assert_all_close(expected_outputs, signature_outputs)
89 expected_outputs = {model.output_names[0]: model(inputs)}
91 self._assert_all_close(expected_outputs, signature_outputs)
128 expected_outputs = {model.output_names[0]: outputs[0],
131 self._assert_all_close(expected_outputs, signature_outputs)
144 expected_outputs = {model.output_names[0]: model(inputs)}
145 self._assert_all_close(expected_outputs, signature_outputs)
167 expected_outputs = {'output_1': model([x, y])}
169 self._assert_all_close(expected_outputs, signature_outputs)
/external/tensorflow/tensorflow/python/kernel_tests/
Dparse_single_example_op_test.py252 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/core/kernels/
Dlookup_table_init_op.cc57 DataTypeVector expected_outputs = {}; in Compute() local
58 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
124 DataTypeVector expected_outputs = {}; in Compute() local
125 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
Dbarrier_ops.cc593 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()
Dlookup_table_op.cc796 DataTypeVector expected_outputs = {table->value_dtype()}; in Compute() local
797 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone_test.cc157 std::vector<int64> expected_outputs; in TEST() member
177 test_case.expected_outputs[num_outputs]); in TEST()
179 EXPECT_EQ(test_case.expected_outputs.size(), num_outputs); in TEST()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dcache_dataset_serialization_test.py52 def expected_outputs(self): member in CacheDatasetSerializationTest
74 self.assertSequenceEqual(outputs, self.expected_outputs())
96 self.assertSequenceEqual(outputs, self.expected_outputs())
126 self.assertSequenceEqual(outputs, self.expected_outputs())
225 self.assertSequenceEqual(outputs, self.expected_outputs())
Dparallel_interleave_dataset_serialization_test.py66 expected_outputs = np.repeat(
74 self.assertSequenceEqual(sorted(actual), expected_outputs)
/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py169 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)
Drandomized_tests.cc915 std::vector<Tensor> expected_outputs, test_outputs; in ExpectTfAndXlaOutputsAreClose() local
918 session_->Run(expected_feeds, expected_fetches, {}, &expected_outputs); in ExpectTfAndXlaOutputsAreClose()
923 for (const Tensor& expected : expected_outputs) { in ExpectTfAndXlaOutputsAreClose()
934 CHECK_EQ(expected_outputs.size(), test_outputs.size()); in ExpectTfAndXlaOutputsAreClose()
936 s = TensorsAreClose(expected_outputs[j], test_outputs[j], atol, rtol); in ExpectTfAndXlaOutputsAreClose()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dbatch_test.py127 expected_outputs = []
131 expected_outputs.append(i * 5 + j)
135 values=expected_outputs,
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc278 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()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc58 const DataTypeSlice expected_outputs, in MatchSignatureHelper() argument
70 if (outputs.size() != expected_outputs.size()) signature_mismatch = true; in MatchSignatureHelper()
72 if (!TypesCompatible(expected_outputs[i], outputs[i])) { in MatchSignatureHelper()
82 DataTypeSliceString(expected_outputs)); in MatchSignatureHelper()
230 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature() argument
231 return MatchSignatureHelper(expected_inputs, expected_outputs, input_types_, in MatchSignature()
873 const DataTypeSlice expected_outputs) { in MatchSignature() argument
879 return MatchSignatureHelper(expected_inputs, expected_outputs, inputs, in MatchSignature()
Dop_kernel.h332 const DataTypeSlice expected_outputs);
1132 const DataTypeSlice expected_outputs);
/external/tensorflow/tensorflow/python/keras/
Dbackend_test.py1220 expected_outputs = inputs_vals.copy()
1223 expected_outputs[1, -mask_last_num_timesteps:] = \
1224 expected_outputs[1, -(mask_last_num_timesteps + 1)]
1245 self.assertAllClose(keras.backend.eval(outputs), expected_outputs)
1266 expected_outputs = np.repeat(inputs_vals[..., None], repeats=2, axis=-1)
1269 expected_outputs[-1, -1] = expected_outputs[-1, -2]
1283 self.assertAllClose(keras.backend.eval(outputs), expected_outputs)
/external/tensorflow/tensorflow/python/ops/
Dnn_test.py1031 actual_outputs, expected_outputs = self.evaluate(
1034 self.assertAllClose(actual_outputs, expected_outputs)
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc529 std::vector<string> expected_outputs{"float: [-1,-1,1]", "float: [2,1,1]", in TEST_F() local
535 EXPECT_EQ(expected_outputs[i], PropToString(prop)); in TEST_F()

12