| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_v_test.cc | 37 const OutputTensors<N>& output_tensors) { in TestSplitVFloat() argument
43 output_dims[i] = IntArrayFromInts(output_tensors.dims[i]); in TestSplitVFloat()
49 (output_tensors.data[i])[j] = 23; in TestSplitVFloat()
72 tensors[3 + i] = CreateTensor(output_tensors.data[i], output_dims[i]); in TestSplitVFloat()
95 TF_LITE_MICRO_EXPECT_NEAR((output_tensors.expected_output_data[i])[j], in TestSplitVFloat()
96 (output_tensors.data[i])[j], 1e-5f); in TestSplitVFloat()
126 tflite::testing::OutputTensors<3> output_tensors; in TF_LITE_MICRO_TEST() local
127 output_tensors.data[0] = output1_data; in TF_LITE_MICRO_TEST()
128 output_tensors.data[1] = output2_data; in TF_LITE_MICRO_TEST()
129 output_tensors.data[2] = output3_data; in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| D | object_detector.cc | 163 const std::vector<const TfLiteTensor*>& output_tensors) { in SanityCheckOutputTensors() argument
164 if (output_tensors.size() != 4) { in SanityCheckOutputTensors()
168 output_tensors.size())); in SanityCheckOutputTensors()
172 if (output_tensors[3]->dims->data[0] != 1) { in SanityCheckOutputTensors()
177 output_tensors[3]->dims->data[0])); in SanityCheckOutputTensors()
180 static_cast<int>(AssertAndReturnTypedTensor<float>(output_tensors[3])[0]); in SanityCheckOutputTensors()
183 if (output_tensors[0]->dims->data[0] != 1 || in SanityCheckOutputTensors()
184 output_tensors[0]->dims->data[1] != num_results || in SanityCheckOutputTensors()
185 output_tensors[0]->dims->data[2] != 4) { in SanityCheckOutputTensors()
191 num_results, output_tensors[0]->dims->data[0], in SanityCheckOutputTensors()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 288 output_tensors = _build_ring_scatter(pred_by_s_d, rank_by_s_d,
291 output_tensors = _strip_padding(output_tensors, pad_len)
293 output_tensors = _reshape_tensors(output_tensors, shape)
294 return output_tensors
474 output_tensors = _build_recursive_hd_scatter(reduced_shards, devices)
476 output_tensors = _reshape_tensors(output_tensors, shape)
477 return output_tensors
586 output_tensors = _build_shuffle_scatter(reduced_shards, dst_devices)
588 output_tensors = _reshape_tensors(output_tensors, shape)
589 return output_tensors
[all …]
|
| D | all_reduce_test.py | 108 output_tensors = ar._build_ring_gather(input_tensors, device_names, 1,
111 self.assertEqual(output_tensors, input_tensors)
115 output_tensors, pad_len = ar._build_ring_gather(
119 self.assertEqual(len(output_tensors), len(input_tensors))
122 for otl in output_tensors:
160 output_tensors = build_f(input_tensors, un_op)
161 sum_reduced = math_ops.add_n(output_tensors)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 142 output_tensors = [o for o in outputs
145 if outputs != output_tensors + output_operations:
150 output_types = [op.dtype for op in output_tensors]
161 if not output_tensors:
162 output_tensors = array_ops.constant(0)
168 output_tensors = control_flow_ops.tuple(output_tensors,
176 output_tensors = tt.trace_tpu(ops.get_default_graph(),
177 output_tensors, None,
179 return output_tensors
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 118 EXPECT_EQ(generated_subgraphs[subgraph_index].output_tensors, in CheckPartitionSubgraphs()
119 expected_subgraphs[subgraph_index].output_tensors); in CheckPartitionSubgraphs()
162 expected_subgraph.output_tensors = {1}; in TEST()
180 expected_subgraph.output_tensors = {1}; in TEST()
200 expected_subgraph.output_tensors = {0}; in TEST()
220 expected_subgraph.output_tensors = {1}; in TEST()
243 expected_subgraph0.output_tensors = {1}; in TEST()
248 expected_subgraph1.output_tensors = {2}; in TEST()
270 expected_subgraph0.output_tensors = {1}; in TEST()
275 expected_subgraph1.output_tensors = {2}; in TEST()
[all …]
|
| D | graph_info.cc | 89 output_subset.output_tensors.push_back(output_index); in Partition()
102 uniquefy(&node_subset.output_tensors); in Partition()
171 input_subset.output_tensors.push_back(input_tensor_index); in UpdateNode()
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla.py | 376 output_tensors, control_deps = _postprocess_flat_outputs(outputs)
378 output_tensors, control_deps = _postprocess_non_flat_outputs(outputs)
380 context.ExitResult(output_tensors)
389 if not output_tensors:
392 output_tensors = [
394 for i, o in enumerate(output_tensors)
400 output_tensors = [
402 for i, o in enumerate(output_tensors)
408 output_tensors = nest.pack_sequence_as(
409 structure=outputs, flat_sequence=output_tensors)
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
184 pywrap_tfe.TFE_Py_TapeSetRecordOperation(op_type, output_tensors,
189 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
192 pywrap_tfe.TFE_Py_TapeSetRecordOperationBackprop(op_type, output_tensors,
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
215 op_type, output_tensors, input_tensors, backward_function,
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 607 def convert(self, graph_def, input_tensors, output_tensors): argument
672 output_tensors=output_tensors,
752 output_tensors = [
765 self).convert(graph_def, input_tensors, output_tensors)
854 output_tensors = [
862 output_tensors)
908 output_tensors = frozen_func.outputs
919 output_tensors,
924 self).convert(graph_def, input_tensors, output_tensors)
977 output_tensors = frozen_func.outputs
[all …]
|
| D | convert.py | 364 output_tensors, argument
522 for output_tensor in output_tensors:
572 input_tensors=[], output_tensors=[], *args, **kwargs)
599 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
625 input_tensors, output_tensors, *args, **kwargs)
662 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
686 return toco_convert_impl(input_data, input_tensors, output_tensors,
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 85 BuildGraph(context, params->input_tensors, params->output_tensors)); in Init()
123 std::vector<hexagon_nn_tensordef> output_tensors; in Eval() local
134 output_tensors.emplace_back(); in Eval()
135 auto& output_tensor = output_tensors.back(); in Eval()
148 output_tensors.data(), output_tensors.size()); in Eval()
241 const TfLiteIntArray* output_tensors) { in BuildGraph() argument
280 TF_LITE_ENSURE_STATUS(builder_->AddOutputTensors(output_tensors, context)); in BuildGraph()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | internal_api_samples.cc | 206 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPI() local
209 output_tensors[i].resize(tensor_ptr->bytes / 4); in RunModelSampleWithInternalAPI()
211 i, CpuMemory{output_tensors[i].data(), tensor_ptr->bytes})); in RunModelSampleWithInternalAPI()
216 CompareCPUGPUResults(cpu_inference.get(), out_refs, output_tensors, 1e-4f); in RunModelSampleWithInternalAPI()
330 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerializedKernels() local
333 output_tensors[i].resize(tensor_ptr->bytes / 4); in RunModelSampleWithInternalAPISerializedKernels()
335 i, CpuMemory{output_tensors[i].data(), tensor_ptr->bytes})); in RunModelSampleWithInternalAPISerializedKernels()
340 CompareCPUGPUResults(cpu_inference.get(), out_refs, output_tensors, 1e-4f); in RunModelSampleWithInternalAPISerializedKernels()
406 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerialized() local
409 output_tensors[i].resize(tensor_ptr->bytes / 4); in RunModelSampleWithInternalAPISerialized()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 120 output_tensors = layer(*args, **kwargs)
123 first_output_tensor = nest.flatten(output_tensors)[0]
128 nest.flatten(node.output_tensors), nest.flatten(output_tensors)):
199 input_tensors, output_tensors, created_layers = (
203 model = Model(input_tensors, output_tensors, name=model.name)
371 if isinstance(cloned_node.output_tensors, list):
372 for j, output_tensor in enumerate(cloned_node.output_tensors):
373 tensor_map[node.output_tensors[j]] = output_tensor
375 tensor_map[node.output_tensors] = cloned_node.output_tensors
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 139 TfLiteIntArrayFree(params.output_tensors); in ~DelegatedInterpreter()
326 params->output_tensors = TfLiteIntArrayCreate(1); in TEST()
327 params->output_tensors->data[0] = 4; in TEST()
385 params->output_tensors = TfLiteIntArrayCreate(1); in TEST()
386 params->output_tensors->data[0] = 4; in TEST()
559 params->output_tensors = TfLiteIntArrayCreate(1); in TEST()
560 params->output_tensors->data[0] = 3; in TEST()
726 params->output_tensors = TfLiteIntArrayCreate(1); in TEST()
727 params->output_tensors->data[0] = 3; in TEST()
735 params->output_tensors = TfLiteIntArrayCreate(1); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | grappler_test.cc | 122 std::vector<Tensor> output_tensors; in EvaluateNodes() local
124 &output_tensors, nullptr)); in EvaluateNodes()
126 return output_tensors; in EvaluateNodes()
139 std::vector<Tensor> output_tensors; in EvaluateFetchNodes() local
141 &output_tensors, nullptr)); in EvaluateFetchNodes()
143 return output_tensors; in EvaluateFetchNodes()
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features.py | 161 output_tensors = []
173 output_tensors.append(processed_tensors)
174 return self._verify_and_concat_tensors(output_tensors)
|
| D | sequence_feature_column.py | 150 output_tensors = []
162 output_tensors.append(self._process_dense_tensor(column, dense_tensor))
170 return self._verify_and_concat_tensors(output_tensors), sequence_length
|
| D | base_feature_layer.py | 115 def _verify_and_concat_tensors(self, output_tensors): argument
118 output_tensors, self._feature_columns)
119 return array_ops.concat(output_tensors, -1)
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 119 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess() argument
121 if (output_tensors.size() != 1) { in Postprocess()
126 output_tensors.size()), in Postprocess()
130 output_tensors, GetMetadataExtractor()->GetOutputTensorMetadata(), in Postprocess()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 263 std::vector<Tensor> output_tensors; in VerifyFusedBatchNorm() local
265 &output_tensors)); in VerifyFusedBatchNorm()
267 *output = output_tensors[0]; in VerifyFusedBatchNorm()
268 *batch_mean = output_tensors[1]; in VerifyFusedBatchNorm()
269 *batch_var = output_tensors[2]; in VerifyFusedBatchNorm()
398 std::vector<Tensor> output_tensors; in VerifyFusedBatchNormGradWithConv2D() local
401 &output_tensors)); in VerifyFusedBatchNormGradWithConv2D()
403 *x_backprop_tensor = output_tensors[0]; in VerifyFusedBatchNormGradWithConv2D()
404 *scale_backprop_tensor = output_tensors[1]; in VerifyFusedBatchNormGradWithConv2D()
405 *offset_backprop_tensor = output_tensors[2]; in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 234 std::vector<tensorflow::Tensor> output_tensors; in TEST() local
236 def, inputs, outputs, false /* initialize_by_zero */, &output_tensors); in TEST()
238 EXPECT_EQ(outputs.size(), output_tensors.size()); in TEST()
239 EXPECT_NEAR(NODE_B_VAL, output_tensors.at(0).scalar<float>()(), in TEST()
241 EXPECT_NEAR(1.0f + NODE_B_VAL, output_tensors.at(1).scalar<float>()(), in TEST()
254 std::vector<tensorflow::Tensor> output_tensors; in TEST() local
256 def, inputs, outputs, true /* initialize_by_zero */, &output_tensors); in TEST()
258 EXPECT_EQ(outputs.size(), output_tensors.size()); in TEST()
259 EXPECT_NEAR(NODE_B_VAL, output_tensors.at(0).scalar<float>()(), in TEST()
261 EXPECT_NEAR(NODE_B_VAL, output_tensors.at(1).scalar<float>()(), in TEST()
[all …]
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 89 std::vector<OutputTensor>* output_tensors) in ProcessOutputs() argument
91 output_tensors->reserve(noutputs); in ProcessOutputs()
102 output_tensors->emplace_back(node, idx); in ProcessOutputs()
168 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs() local
170 outputs, &output_tensors); in TF_GraphToFunctionWithControlOutputs()
211 output_tensors, output_names_vec, control_output_nodes, in TF_GraphToFunctionWithControlOutputs()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 158 const string& op_type, const std::vector<TapeTensor>& output_tensors,
283 const std::vector<TapeTensor>& output_tensors,
331 const string& op_type, const std::vector<TapeTensor>& output_tensors,
407 const string& op_type, const std::vector<TapeTensor>& output_tensors, in RecordOperation() argument
423 tensors.reserve(output_tensors.size()); in RecordOperation()
424 for (const TapeTensor& o : output_tensors) { in RecordOperation()
925 const string& op_type, const std::vector<TapeTensor>& output_tensors, in ForwardpropFromTape() argument
947 sources.reserve(output_tensors.size()); in ForwardpropFromTape()
948 for (const TapeTensor& output_tensor : output_tensors) { in ForwardpropFromTape()
1024 const std::vector<TapeTensor>& output_tensors, in Accumulate() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.h | 50 void AddOutputTensors(const TfLiteIntArray* output_tensors,
54 void AddInputTensors(const TfLiteIntArray* output_tensors,
|