Searched refs:sub_output (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | signature_runner_test.cc | 84 const TfLiteTensor* sub_output = sub_runner->output_tensor("output_0"); in TEST() local
86 ASSERT_NE(sub_output, nullptr); in TEST()
91 ASSERT_EQ(sub_output->data.f[0], -1); in TEST()
92 ASSERT_EQ(sub_output->data.f[1], 1); in TEST()
93 ASSERT_EQ(sub_output->data.f[2], 3); in TEST()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_signature_runner_test.cc | 111 const TfLiteTensor* sub_output = in TEST() local
115 ASSERT_NE(sub_output, nullptr); in TEST()
120 ASSERT_EQ(sub_output->data.f[0], -1); in TEST()
121 ASSERT_EQ(sub_output->data.f[1], 1); in TEST()
122 ASSERT_EQ(sub_output->data.f[2], 3); in TEST()
|
| /external/executorch/backends/arm/operators/ |
| D | op_sub.py | 48 sub_output = tosa_graph.addIntermediate(broadcasted_shape, ts.DType.INT32)
57 [sub_output.name],
61 tqutils.rescale_node_back_to_int8(node, sub_output, scale, tosa_graph)
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | gpu_model_test_util.cc | 737 auto sub_output = graph.NewValue(); in TestLinkingConvolutionFirstTanh2InputDiff() local
738 sub_output->tensor.type = DataType::FLOAT32; in TestLinkingConvolutionFirstTanh2InputDiff()
739 sub_output->tensor.shape = BHWC(1, 32, 32, 16); in TestLinkingConvolutionFirstTanh2InputDiff()
743 RETURN_IF_ERROR(graph.SetProducer(sub_node->id, sub_output->id)); in TestLinkingConvolutionFirstTanh2InputDiff()
805 sub_output->tensor.shape, &dst_tensor_v0)); in TestLinkingConvolutionFirstTanh2InputDiff()
855 auto sub_output = graph.NewValue(); in TestLinkingConvolutionSecondTanh2InputDiff() local
856 sub_output->tensor.type = DataType::FLOAT32; in TestLinkingConvolutionSecondTanh2InputDiff()
857 sub_output->tensor.shape = BHWC(1, 32, 32, 16); in TestLinkingConvolutionSecondTanh2InputDiff()
861 RETURN_IF_ERROR(graph.SetProducer(sub_node->id, sub_output->id)); in TestLinkingConvolutionSecondTanh2InputDiff()
923 sub_output->tensor.shape, &dst_tensor_v0)); in TestLinkingConvolutionSecondTanh2InputDiff()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | mean_stddev_normalization.cc | 375 auto sub_output = graph.FindOutputs(sub_node->id)[0]->id; in TryMeanStdDevNormalization() local
376 auto consumers = graph.FindConsumers(sub_output); in TryMeanStdDevNormalization()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | autograd_not_implemented_fallback.cpp | 591 for (auto& sub_output : aliased_output) { in autogradNotImplementedInplaceOrViewFallbackImpl() local
594 /* tensor=*/sub_output, in autogradNotImplementedInplaceOrViewFallbackImpl()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 245 sub_output = sub_signature_runner(x=input_data)
246 self.assertEqual(sub_output['output_0'], -2)
2042 sub_output = sub_signature_runner(x=input_data)
2043 self.assertEqual(sub_output['output_0'], -2)
2139 sub_output = sub_signature_runner(x=input_data)
2140 self.assertIsNotNone(sub_output['output_0'])
|