Home
last modified time | relevance | path

Searched refs:outputs (Results 1 – 25 of 1773) sorted by relevance

12345678910>>...71

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dcache_dataset_serialization_test.py63 outputs = self.gen_outputs(ds_fn, [], 5, verify_exhausted=False)
64 self.assertSequenceEqual(outputs, range(5))
68 outputs.extend(
74 self.assertSequenceEqual(outputs, self.expected_outputs())
84 outputs = self.gen_outputs(
86 self.assertSequenceEqual(outputs, range(8))
89 outputs = outputs[:5]
90 outputs.extend(
96 self.assertSequenceEqual(outputs, self.expected_outputs())
115 outputs = self.gen_outputs(ds_fn, [], 15, verify_exhausted=False)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dunicode_transcode_op_test.py39 outputs = string_ops.unicode_transcode(
46 values = self.evaluate(outputs)
49 outputs = string_ops.unicode_transcode(
56 values = self.evaluate(outputs)
59 outputs = string_ops.unicode_transcode(
66 values = self.evaluate(outputs)
74 outputs = string_ops.unicode_transcode(
81 values = self.evaluate(outputs)
87 outputs = string_ops.unicode_transcode(
94 values = self.evaluate(outputs)
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_dilated_conv.cc65 CHECK_EQ(conv_op->outputs.size(), 1); in ResolveDilatedConv()
68 auto* post_conv_op = GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
78 CHECK_EQ(post_conv_op->outputs.size(), 1); in ResolveDilatedConv()
83 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
84 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
89 CHECK_EQ(pad_op->outputs.size(), 1); in ResolveDilatedConv()
95 ? GetOpWithInput(*model, pad_op->outputs[0]) in ResolveDilatedConv()
97 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
98 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
103 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv()
[all …]
Didentify_lstm.cc153 if (model->GetArray(final_output_mul->outputs[0]).data_type == in Run()
264 lstm_cell_op->outputs.resize(LstmCellOperator::NUM_OUTPUTS); in Run()
265 lstm_cell_op->outputs[LstmCellOperator::STATE_OUTPUT] = in Run()
267 lstm_cell_op->outputs[LstmCellOperator::ACTIV_OUTPUT] = in Run()
268 final_output_mul->outputs[0]; in Run()
275 lstm_cell_op->outputs[LstmCellOperator::STATE_OUTPUT], in Run()
276 lstm_cell_op->outputs[LstmCellOperator::ACTIV_OUTPUT])); in Run()
281 model->GetArray(concat_inputs->outputs[0]).data_type; in Run()
282 lstm_cell_op->outputs[LstmCellOperator::CONCAT_TEMP] = concat_temp_array_name; in Run()
287 model->GetArray(fully_connected->outputs[0]).data_type; in Run()
[all …]
Dunroll_batch_matmul.cc38 absl::StrCat(batch_op->outputs[0], "_b", absl::StrJoin(batch, "-")); in UnrollBatchMatMul3D()
57 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")}; in UnrollBatchMatMul3D()
58 auto& slice_a_op_output = model->GetOrCreateArray(slice_a_op->outputs[0]); in UnrollBatchMatMul3D()
65 slice_a_op->outputs[0], in UnrollBatchMatMul3D()
68 slice_a_reshape_op->outputs = { in UnrollBatchMatMul3D()
71 model->GetOrCreateArray(slice_a_reshape_op->outputs[0]); in UnrollBatchMatMul3D()
89 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")}; in UnrollBatchMatMul3D()
90 auto& slice_b_op_output = model->GetOrCreateArray(slice_b_op->outputs[0]); in UnrollBatchMatMul3D()
97 slice_b_op->outputs[0], in UnrollBatchMatMul3D()
100 slice_b_reshape_op->outputs = { in UnrollBatchMatMul3D()
[all …]
Dpropagate_array_data_types.cc29 for (const auto& output : op->outputs) { in SetDataTypeForAllOutputs()
52 for (const auto& output : op->outputs) { in Run()
105 CHECK_EQ(op->outputs.size(), 1); in Run()
107 model->GetArray(op->outputs[0]).data_type = cast_op->dst_data_type; in Run()
112 CHECK_EQ(op->outputs.size(), 1); in Run()
114 model->GetArray(op->outputs[0]).data_type = argmax_op->output_data_type; in Run()
119 CHECK_EQ(op->outputs.size(), 1); in Run()
121 model->GetArray(op->outputs[0]).data_type = argmin_op->output_data_type; in Run()
136 CHECK_EQ(op->outputs.size(), 1); in Run()
146 CHECK_EQ(op->outputs.size(), 1); in Run()
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dtensorflow_op_layer_test.py40 outputs = gen_nn_ops.relu(x)
41 return inputs, outputs
47 outputs = array_ops.identity(x)
48 assert 'Identity' in outputs.name
49 return inputs, outputs
56 outputs = gen_nn_ops.relu(x)
57 return inputs, outputs
64 outputs = keras.layers.Dense(10)(x)
65 return inputs, outputs
73 outputs = keras.layers.Dense(10)(x)
[all …]
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc38 std::vector<Tensor> outputs; in TEST() local
40 TF_EXPECT_OK(session.Run({c}, &outputs)); in TEST()
41 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({1, 1}, {1, 2})); in TEST()
50 std::vector<Tensor> outputs; in TEST() local
52 TF_EXPECT_OK(session.Run({{a, 1}, {b, 41}}, {c}, &outputs)); in TEST()
53 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({42}, {})); in TEST()
61 std::vector<Tensor> outputs; in TEST() local
63 TF_EXPECT_OK(session.Run({{a, {1, 1}}}, {c}, &outputs)); in TEST()
64 test::ExpectTensorEqual<int>(outputs[0], test::AsTensor<int>({3, 3}, {2})); in TEST()
67 outputs.clear(); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
DNativeInterpreterWrapperTest.java119 Map<Integer, Object> outputs = new HashMap<>(); in testRunWithFloat() local
120 outputs.put(0, parsedOutputs); in testRunWithFloat()
121 wrapper.run(inputs, outputs); in testRunWithFloat()
138 Map<Integer, Object> outputs = new HashMap<>(); in testRunWithBufferOutput() local
139 outputs.put(0, parsedOutput); in testRunWithBufferOutput()
140 wrapper.run(inputs, outputs); in testRunWithBufferOutput()
158 Map<Integer, Object> outputs = new HashMap<>(); in testRunWithInputsOfSameDims() local
159 outputs.put(0, parsedOutputs); in testRunWithInputsOfSameDims()
160 wrapper.run(inputs, outputs); in testRunWithInputsOfSameDims()
165 outputs.put(0, parsedOutputs); in testRunWithInputsOfSameDims()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc133 std::vector<Tensor> outputs; in TEST_F() local
134 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
137 ASSERT_EQ(1, outputs.size()); in TEST_F()
140 auto mat = outputs[0].matrix<float>(); in TEST_F()
141 ASSERT_TRUE(outputs[0].IsInitialized()); in TEST_F()
159 std::vector<Tensor> outputs; in TEST_F() local
160 TF_ASSERT_OK(session->RunCallable(handle, {}, &outputs, nullptr)); in TEST_F()
162 ASSERT_EQ(1, outputs.size()); in TEST_F()
165 auto mat = outputs[0].matrix<float>(); in TEST_F()
166 ASSERT_TRUE(outputs[0].IsInitialized()); in TEST_F()
[all …]
Dgraph_runner_test.cc50 std::vector<Tensor> outputs; in TEST() local
51 Status s = graph_runner.Run(root.graph(), nullptr, {}, {c.name()}, &outputs); in TEST()
53 ExpectEqual(42.0f, outputs[0].scalar<float>()()); in TEST()
73 std::vector<Tensor> outputs; in TEST() local
77 graph_runner.Run(root.graph(), nullptr, inputs, {"add:0"}, &outputs); in TEST()
80 ExpectEqual(3.0f, outputs[0].scalar<float>()()); in TEST()
88 std::vector<Tensor> outputs; in TEST() local
90 &outputs); in TEST()
92 ExpectEqual(42.0f, outputs[0].scalar<float>()()); in TEST()
93 ExpectEqual(3.14f, outputs[1].scalar<float>()()); in TEST()
[all …]
/external/mesa3d/src/gallium/drivers/r300/
Dr300_vs.c109 struct r300_shader_semantics* outputs = &vs->outputs; in set_vertex_inputs_outputs() local
112 boolean any_bcolor_used = outputs->bcolor[0] != ATTR_UNUSED || in set_vertex_inputs_outputs()
113 outputs->bcolor[1] != ATTR_UNUSED; in set_vertex_inputs_outputs()
120 if (outputs->pos != ATTR_UNUSED) { in set_vertex_inputs_outputs()
121 c->code->outputs[outputs->pos] = reg++; in set_vertex_inputs_outputs()
127 if (outputs->psize != ATTR_UNUSED) { in set_vertex_inputs_outputs()
128 c->code->outputs[outputs->psize] = reg++; in set_vertex_inputs_outputs()
140 if (outputs->color[i] != ATTR_UNUSED) { in set_vertex_inputs_outputs()
141 c->code->outputs[outputs->color[i]] = reg++; in set_vertex_inputs_outputs()
143 outputs->color[1] != ATTR_UNUSED) { in set_vertex_inputs_outputs()
[all …]
/external/tensorflow/tensorflow/contrib/specs/python/
Dspecs_test.py44 outputs = specs.create_net(spec, inputs)
45 self.assertEqual(outputs.get_shape().as_list(), [1, 18, 19, 64])
47 result = outputs.eval()
59 outputs = specs.create_net(spec, inputs)
60 self.assertEqual(outputs.get_shape().as_list(), [17, 55])
62 result = outputs.eval()
69 outputs = specs.create_net(spec, inputs)
70 self.assertEqual(outputs.get_shape().as_list(), [17, 10])
72 result = outputs.eval()
83 outputs = specs.create_net(spec, inputs)
[all …]
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py152 outputs = body(*_pack_sequence_as(orig_loop_vars, args))
153 if not nest.is_sequence(outputs):
154 outputs = [outputs]
157 nest.assert_same_structure(list(outputs), list(orig_loop_vars))
159 outputs = _tensor_array_to_flow(outputs)
163 return [loop_counter + 1, maximum_iterations_arg] + list(outputs)
181 body_graph.outputs.extend(body_graph.internal_captures)
200 body_graph.outputs[first_loop_var_index:first_loop_var_index +
213 outputs = gen_functional_ops._while(
217 output_shapes=[t.shape for t in body_graph.outputs],
[all …]
Dcudnn_rnn_grad.py35 output=op.outputs[0],
36 output_h=op.outputs[1],
37 output_c=op.outputs[2],
41 reserve_space=op.outputs[3],
60 output=op.outputs[0],
61 output_h=op.outputs[1],
62 output_c=op.outputs[2],
66 reserve_space=op.outputs[3],
67 host_reserved=op.outputs[4],
89 output=op.outputs[0],
[all …]
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop.cc35 std::vector<OutputTensor> ToOutputTensors(const std::vector<Output>& outputs) { in ToOutputTensors() argument
36 std::vector<OutputTensor> result(outputs.size()); in ToOutputTensors()
37 for (int i = 0; i < outputs.size(); ++i) { in ToOutputTensors()
38 result[i] = ToOutputTensor(outputs[i]); in ToOutputTensors()
44 std::vector<Node*> ToNodes(const std::vector<Output>& outputs) { in ToNodes() argument
45 std::vector<Node*> result(outputs.size()); in ToNodes()
46 for (int i = 0; i < outputs.size(); ++i) { in ToNodes()
47 result[i] = outputs[i].node(); in ToNodes()
123 std::vector<Output>* outputs) { in CreateBody() argument
124 DCHECK(outputs != nullptr); in CreateBody()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc184 std::vector<Tensor> outputs; in TEST() local
185 TF_CHECK_OK(session.Run({create_const}, &outputs)); in TEST()
186 EXPECT_EQ(1, outputs.size()); in TEST()
187 EXPECT_EQ(DT_VARIANT, outputs[0].dtype()); in TEST()
188 EXPECT_EQ(0, outputs[0].dims()); in TEST()
189 const Variant& variant = outputs[0].scalar<Variant>()(); in TEST()
218 std::vector<Tensor> outputs; in TEST() local
222 TF_CHECK_OK(session.Run({create_const}, &outputs)); in TEST()
229 EXPECT_EQ(1, outputs.size()); in TEST()
230 EXPECT_EQ(DT_VARIANT, outputs[0].dtype()); in TEST()
[all …]
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dhelper.py158 def sample(self, time, outputs, state, name=None): argument
163 def next_inputs(self, time, outputs, state, sample_ids, name=None): argument
215 def sample(self, time, outputs, state, name=None): argument
217 name, "%sSample" % type(self).__name__, (time, outputs, state)):
218 return self._sample_fn(time=time, outputs=outputs, state=state)
220 def next_inputs(self, time, outputs, state, sample_ids, name=None): argument
222 name, "%sNextInputs" % type(self).__name__, (time, outputs, state)):
224 time=time, outputs=outputs, state=state, sample_ids=sample_ids)
294 def sample(self, time, outputs, name=None, **unused_kwargs): argument
295 with ops.name_scope(name, "TrainingHelperSample", [time, outputs]):
[all …]
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc109 xla::Literal ReadOutputLiteral(const std::vector<Tensor>& outputs, size_t idx) { in ReadOutputLiteral() argument
111 CHECK(response.ParseFromString(outputs[idx].scalar<string>()())); in ReadOutputLiteral()
263 std::vector<Tensor> outputs; in TEST() local
264 TF_EXPECT_OK(session.Run({read_back}, &outputs)); in TEST()
265 EXPECT_EQ(outputs.size(), 1); in TEST()
268 EXPECT_TRUE(response.ParseFromString(outputs[0].scalar<string>()())); in TEST()
294 std::vector<Tensor> outputs; in TEST() local
295 TF_EXPECT_OK(session.Run({read_back}, &outputs)); in TEST()
296 EXPECT_EQ(outputs.size(), 1); in TEST()
299 EXPECT_TRUE(response.ParseFromString(outputs[0].scalar<string>()())); in TEST()
[all …]
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py39 def build_signature_def(inputs=None, outputs=None, method_name=None): argument
56 if outputs is not None:
57 for item in outputs:
58 signature_def.outputs[item].CopyFrom(outputs[item])
178 def predict_signature_def(inputs, outputs): argument
197 if outputs is None or not outputs:
203 for key, tensor in outputs.items()}
290 if not signature_def.outputs.keys():
307 if (set(signature_def.outputs.keys())
310 if (signature_def.outputs[signature_constants.REGRESS_OUTPUTS].dtype !=
[all …]
Dsignature_def_utils_test.py48 def _make_signature(inputs, outputs, name=None): argument
55 for output_name, tensor in outputs.items()
72 outputs = dict()
73 outputs["foo-output"] = y_tensor_info
76 inputs, outputs, "foo-method-name")
88 self.assertEqual(1, len(signature_def.outputs))
89 y_tensor_info_actual = signature_def.outputs["foo-output"]
113 self.assertEqual(1, len(signature_def.outputs))
115 signature_def.outputs[signature_constants.REGRESS_OUTPUTS])
140 self.assertEqual(2, len(signature_def.outputs))
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py121 outputs = body(*(inputs + dequeue_ops))
124 if not isinstance(outputs, (list, tuple)):
125 outputs = (outputs,)
127 outputs = [
129 for o in outputs
133 output_operations = [o for o in outputs if isinstance(o, ops.Operation)]
134 output_tensors = [o for o in outputs
137 if outputs != output_tensors + output_operations:
215 outputs = while_loop(
217 outputs = _convert_to_list(outputs)
[all …]
/external/tensorflow/tensorflow/core/graph/
Dquantize_training_test.cc405 std::vector<Tensor> outputs; in TEST_F() local
406 TF_ASSERT_OK(sess->Run({}, {min_const_name, max_const_name}, {}, &outputs)); in TEST_F()
407 EXPECT_EQ(outputs[0].flat<float>()(0), 0.0); in TEST_F()
408 EXPECT_EQ(outputs[1].flat<float>()(0), 6.0); in TEST_F()
415 TF_ASSERT_OK(sess->Run({{"a", a1}}, {m1->name()}, {}, &outputs)); in TEST_F()
421 TF_ASSERT_OK(sess->Run({}, {min_var_name, max_var_name}, {}, &outputs)); in TEST_F()
422 EXPECT_EQ(outputs[0].flat<float>()(0), 0.0); in TEST_F()
423 EXPECT_EQ(outputs[1].flat<float>()(0), 3.0); in TEST_F()
426 TF_ASSERT_OK(sess->Run({}, {min_const_name, max_const_name}, {}, &outputs)); in TEST_F()
427 EXPECT_EQ(outputs[0].flat<float>()(0), 0.0); in TEST_F()
[all …]
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DSessionTest.java37 TestUtil.AutoCloseableList<Tensor<?>> outputs = in runUsingOperationNames() argument
39 assertEquals(1, outputs.size()); in runUsingOperationNames()
41 assertArrayEquals(expected, outputs.get(0).copyTo(new int[1][1])); in runUsingOperationNames()
54 TestUtil.AutoCloseableList<Tensor<?>> outputs = in runUsingOperationHandles() argument
56 assertEquals(1, outputs.size()); in runUsingOperationHandles()
58 assertArrayEquals(expected, outputs.get(0).copyTo(new int[1][1])); in runUsingOperationHandles()
113 …TestUtil.AutoCloseableList<Tensor<?>> outputs = new TestUtil.AutoCloseableList<Tensor<?>>(result.o… in runWithMetadata() local
114 assertEquals(1, outputs.size()); in runWithMetadata()
116 assertArrayEquals(expected, outputs.get(0).copyTo(new int[1][1])); in runWithMetadata()
125 outputs.close(); in runWithMetadata()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc71 const std::vector<int>& outputs, TfLiteType type, in AddTensors() argument
78 if (std::find(outputs.begin(), outputs.end(), i) != outputs.end()) { in AddTensors()
88 CHECK_EQ(interpreter_->SetOutputs(outputs), kTfLiteOk); in AddTensors()
92 const std::vector<int>& outputs) { in AddTfLiteMulOp() argument
99 auto* o = &context->tensors[node->outputs->data[0]]; in AddTfLiteMulOp()
105 auto* o = &context->tensors[node->outputs->data[0]]; in AddTfLiteMulOp()
112 CHECK_EQ(interpreter_->AddNodeWithParameters(inputs, outputs, nullptr, 0, in AddTfLiteMulOp()
118 const std::vector<int>& outputs) { in AddTfOp() argument
147 AddTfOp("FlexUnpack", "Unpack", attributes, inputs, outputs); in AddTfOp()
150 AddTfOp("FlexIdentity", "Identity", attributes, inputs, outputs); in AddTfOp()
[all …]

12345678910>>...71