Searched refs:outputs_size (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax_test.cc | 37 constexpr int outputs_size = 1; in TestSoftmaxFloat() local
38 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxFloat()
101 constexpr int outputs_size = 1; in TestSoftmaxQuantized() local
102 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxQuantized()
|
| D | depthwise_conv_test.cc | 44 constexpr int outputs_size = 1; in TestDepthwiseConvFloat() local
45 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvFloat()
125 constexpr int outputs_size = 1; in TestDepthwiseConvQuantized() local
126 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvQuantized()
|
| D | fully_connected_test.cc | 44 constexpr int outputs_size = 1; in TestFullyConnectedFloat() local
45 constexpr int tensors_size = inputs_size + outputs_size; in TestFullyConnectedFloat()
118 constexpr int outputs_size = 1; in TestFullyConnectedQuantized() local
119 constexpr int tensors_size = inputs_size + outputs_size; in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter_test.cc | 125 constexpr size_t outputs_size = 1; in BuildMockModel() local
126 const int32_t outputs[outputs_size] = {2}; in BuildMockModel()
140 builder.CreateVector(outputs, outputs_size), in BuildMockModel()
174 TF_LITE_MICRO_EXPECT_EQ(1, interpreter.outputs_size()); in TF_LITE_MICRO_TEST()
|
| D | micro_interpreter.h | 49 size_t outputs_size() const { return subgraph_->outputs()->Length(); } in outputs_size() function
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 179 const int outputs_size);
198 const int outputs_size, const bool append_input_params,
|
| D | graph_transferer.cc | 857 const int outputs_size) { in AppendNodeParams() argument
866 node_info.set_output_count(static_cast<int>(outputs_size)); in AppendNodeParams()
949 const int outputs_size, const bool append_input_params, in AppendNodeParamsWithIoParams() argument
959 outputs_size); in AppendNodeParamsWithIoParams()
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim_test.cc | 88 EXPECT_EQ(1, regression_signature_def.outputs_size()); in LoadAndValidateSavedModelBundle()
142 EXPECT_EQ(1, signature_def.outputs_size()); in TEST()
550 EXPECT_EQ(1, signature_def.outputs_size()); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parallel_interleave_dataset_op.cc | 860 int64 outputs_size; in ReadWorkerStateLocked() local
863 &outputs_size)); in ReadWorkerStateLocked()
864 for (int i = 0; i < outputs_size; ++i) { in ReadWorkerStateLocked()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 457 if (op_info.outputs_size() > 0) { in PredictCwiseOp()
938 if (!shape_found && op_info.outputs_size() == 1) { in CountConv2DBackpropInputOperations()
990 if (!shape_found && op_info.outputs_size() == 1) { in CountConv2DBackpropFilterOperations()
1339 if (op_info.outputs_size() == 0 || op_info.inputs_size() < inputs_needed) { in PredictGatherOrSlice()
1608 if (!shape_found && op_info.outputs_size() > 0) { in PredictAvgPoolGrad()
|
| D | virtual_scheduler_test.cc | 2382 EXPECT_EQ(1, ops_executed["x"].op_info.outputs_size()); in TEST_F()
2383 EXPECT_EQ(1, ops_executed["y"].op_info.outputs_size()); in TEST_F()
2384 EXPECT_EQ(1, ops_executed["f"].op_info.outputs_size()); in TEST_F()
|