| /external/tensorflow/tensorflow/contrib/mixed_precision/python/ |
| loss_scale_manager_test.py | 60 expected_outputs,
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/ |
| sparse_tensor_slice_dataset_op_test.cc | 70 std::vector<SparseTensorParam> expected_outputs; member in struct:tensorflow::data::__anon45334::TestCase
134 std::vector<SparseTensorParam> expected_outputs = local
135 test_case.expected_outputs;
166 ExpectEqual(out_tensors[0], expected_outputs[cur_slice].indices));
168 ExpectEqual(out_tensors[1], expected_outputs[cur_slice].values));
170 expected_outputs[cur_slice].dense_shape));
209 std::vector<SparseTensorParam> expected_outputs = local
210 test_case.expected_outputs;
228 expected_outputs[0].indices.dtype(), expected_outputs[0].values.dtype()
242 std::vector<SparseTensorParam> expected_outputs = local
275 std::vector<SparseTensorParam> expected_outputs = local
334 std::vector<SparseTensorParam> expected_outputs = local
373 std::vector<SparseTensorParam> expected_outputs = local
442 std::vector<SparseTensorParam> expected_outputs = local
[all...] |
| tensor_slice_dataset_op_test.cc | 73 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45340::TestCase
95 {"b"})}}, // expected_outputs
125 TensorShape({3}), {4, 5, 6})}}, // expected_outputs
134 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
148 shapes.emplace_back(expected_outputs[i].shape());
179 expected_outputs.size());
186 expected_outputs[i + num_tensors_per_slice * cur_slice]
193 expected_outputs[i + num_tensors_per_slice * cur_slice]));
233 std::vector<Tensor> expected_outputs = test_case.expected_outputs local
276 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
320 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
391 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
440 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
523 std::vector<Tensor> expected_outputs = test_case.expected_outputs; local
[all...] |
| zip_dataset_op_test.cc | 84 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45345::TestParam
92 /*expected_outputs*/
106 /*expected_outputs*/
171 auto expected_outputs_it = test_case.expected_outputs.begin();
179 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end());
185 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end());
208 expected_output_dtypes.emplace_back(test_case.expected_outputs[i].dtype());
226 expected_output_shapes.emplace_back(test_case.expected_outputs[i].shape());
242 test_case.expected_outputs.size() / num_tensors_per_slice);
270 expected_output_dtypes.emplace_back(test_case.expected_outputs[i].dtype())
[all...] |
| repeat_dataset_op_test.cc | 67 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45327::TestCase
80 /*expected_outputs*/
102 /*expected_outputs*/
115 /*expected_outputs*/
164 auto expected_outputs_it = test_case.expected_outputs.begin();
179 if (expected_outputs_it == test_case.expected_outputs.end()) {
180 expected_outputs_it = test_case.expected_outputs.begin();
191 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end());
197 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end());
478 auto expected_outputs_it = test_case.expected_outputs.begin()
[all...] |
| concatenate_dataset_op_test.cc | 74 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45261::TestParam
91 /*expected_outputs*/
118 /*expected_outputs*/
186 auto expected_outputs_it = test_case.expected_outputs.begin();
194 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end());
200 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end());
212 /*expected_outputs*/ {},
308 auto expected_outputs_it = test_case.expected_outputs.begin();
331 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end());
341 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end())
[all...] |
| tensor_dataset_op_test.cc | 71 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45338::TestCase
87 /*expected_outputs*/
113 /*expected_outputs*/
169 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size());
176 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>();
179 TF_EXPECT_OK(ExpectEqual(out_tensors[i], test_case.expected_outputs[i]));
501 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size());
509 test_case.expected_outputs[i].scalar<Variant>()().get<Tensor>();
513 ExpectEqual(out_tensors[i], test_case.expected_outputs[i]));
|
| take_dataset_op_test.cc | 64 std::vector<Tensor> expected_outputs; member in struct:tensorflow::data::__anon45336::TestCase
77 /*expected_outputs*/
95 /*expected_outputs*/
119 /*expected_outputs*/
143 /*expected_outputs*/
190 auto expected_outputs_it = test_case.expected_outputs.begin();
198 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end());
204 EXPECT_EQ(expected_outputs_it, test_case.expected_outputs.end());
486 auto expected_outputs_it = test_case.expected_outputs.begin();
536 EXPECT_NE(expected_outputs_it, test_case.expected_outputs.end())
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| standalone_test.cc | 157 std::vector<int64> expected_outputs; member in struct:tensorflow::data::standalone::__anon44812::TestCase
177 test_case.expected_outputs[num_outputs]);
179 EXPECT_EQ(test_case.expected_outputs.size(), num_outputs);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| lookup_table_init_op.cc | 57 DataTypeVector expected_outputs = {}; variable
58 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
124 DataTypeVector expected_outputs = {}; variable
125 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
|
| barrier_ops.cc | 593 DataTypeVector expected_outputs = {DT_INT64, DT_STRING}; variable
595 expected_outputs.push_back(dt);
598 ctx, ctx->MatchSignature(expected_inputs, expected_outputs), callback);
|
| lookup_table_op.cc | 796 DataTypeVector expected_outputs = {table->value_dtype()}; variable
797 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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/python/keras/saving/ |
| saving_utils_test.py | 72 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
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| cache_dataset_serialization_test.py | 52 def expected_outputs(self): member in class: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())
|
| parallel_interleave_dataset_serialization_test.py | 66 expected_outputs = np.repeat(
74 self.assertSequenceEqual(sorted(actual), expected_outputs)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| 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)
|
| randomized_tests.cc | 915 std::vector<Tensor> expected_outputs, test_outputs; local
918 session_->Run(expected_feeds, expected_fetches, {}, &expected_outputs);
923 for (const Tensor& expected : expected_outputs) {
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| batch_test.py | 127 expected_outputs = []
131 expected_outputs.append(i * 5 + j)
135 values=expected_outputs,
|
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model_test.cc | 278 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; local
280 BuildSignatureDef(expected_inputs, expected_outputs);
289 EXPECT_EQ(expected_outputs, outputs);
308 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; local
310 EXPECT_EQ(expected_outputs, outputs);
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.cc | 58 const DataTypeSlice expected_outputs,
70 if (outputs.size() != expected_outputs.size()) signature_mismatch = true;
72 if (!TypesCompatible(expected_outputs[i], outputs[i])) {
82 DataTypeSliceString(expected_outputs));
230 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) {
231 return MatchSignatureHelper(expected_inputs, expected_outputs, input_types_,
[all...] |
| op_kernel.h | 328 // If expected_inputs == inputs() and expected_outputs == output_types(),
332 const DataTypeSlice expected_outputs);
[all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| backend_test.py | [all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties_test.cc | 529 std::vector<string> expected_outputs{"float: [-1,-1,1]", "float: [2,1,1]",
535 EXPECT_EQ(expected_outputs[i], PropToString(prop));
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_test.py | [all...] |
