Home
last modified time | relevance | path

Searched refs:add_outputs (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/evaluation/
Devaluation_stage_test.cc44 config.add_outputs(kOutputMapping); in GetIdentityStageConfig()
63 config.add_outputs(kOutputMapping); in TEST()
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py238 output, = custom.add_outputs(output)
265 return custom.add_outputs(a * x + b, x)
291 output, = custom.add_outputs(output)
332 custom.add_outputs([res])
Dop_hint.py445 def add_outputs(self, *args, **kwargs): member in OpHint
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc102 auto output = op_info->add_outputs(); in DescribeArbitraryRankOutput()
249 DescribeTensor4D(size1, 1, 1, 1, op_context.op_info.add_outputs()); in DescribeUnaryOp()
268 DescribeTensor4D(2 * size1, size2, 1, 1, op_context.op_info.add_outputs()); in DescribeBinaryOp()
284 DescribeTensor4D(1, 1, size2, size1, op_context.op_info.add_outputs()); in DescribeBiasAdd()
382 DescribeTensor4D(y[0], y[1], y[2], y[3], op_info.add_outputs()); in DescribePoolingOp()
390 DescribeTensor4D(x[0], x[1], x[2], x[3], op_info.add_outputs()); in DescribePoolingOp()
396 DescribeTensor4D(x[0], x[1], x[2], x[3], op_info.add_outputs()); in DescribePoolingOp()
432 auto* tensor = op_info.add_outputs(); in DescribeFusedBatchNorm()
Dvirtual_scheduler.cc642 *op_info.add_outputs() = output; in GetCurrNode()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-op-hint.pbtxt74 name: "add_outputs"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.-op-hint.pbtxt74 name: "add_outputs"
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule.cc46 op_context.op_info.add_outputs()->Swap(&output); in PredictExecutionTime()
/external/tensorflow/tensorflow/lite/schema/
Dschema_generated.h8354 void add_outputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> outputs) {
8398 builder_.add_outputs(outputs);
8495 void add_outputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> outputs) {
8526 builder_.add_outputs(outputs);