| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 89 const size_t inputs_size = PyList_Size(input_value); in FeedTensor() local
91 if (inputs_size != interpreter_->inputs().size()) { in FeedTensor()
94 interpreter_->inputs().size(), inputs_size); in FeedTensor()
98 for (size_t i = 0; i < inputs_size; i++) { in FeedTensor()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 73 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
74 inputs_size = inputs_size or state_size
76 random_ops.random_normal([batch_size, inputs_size]) for _ in range(steps)
112 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
114 inputs_size=inputs_size)
186 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
187 inputs_size = inputs_size or state_size
189 random_ops.random_normal([batch_size, inputs_size]) for _ in range(steps)
496 inputs_size=128)
599 100, 32, 8, inputs_size=128)
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax_test.cc | 36 constexpr int inputs_size = 2; in TestSoftmaxFloat() local
38 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxFloat()
100 constexpr int inputs_size = 1; in TestSoftmaxQuantized() local
102 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxQuantized()
|
| D | depthwise_conv_test.cc | 43 constexpr int inputs_size = 3; in TestDepthwiseConvFloat() local
45 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvFloat()
124 constexpr int inputs_size = 3; in TestDepthwiseConvQuantized() local
126 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvQuantized()
|
| D | fully_connected_test.cc | 43 constexpr int inputs_size = 3; in TestFullyConnectedFloat() local
45 constexpr int tensors_size = inputs_size + outputs_size; in TestFullyConnectedFloat()
117 constexpr int inputs_size = 3; 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 | 123 constexpr size_t inputs_size = 1; in BuildMockModel() local
124 const int32_t inputs[inputs_size] = {0}; in BuildMockModel()
139 builder.CreateVector(inputs, inputs_size), in BuildMockModel()
173 TF_LITE_MICRO_EXPECT_EQ(1, interpreter.inputs_size()); in TF_LITE_MICRO_TEST()
|
| D | micro_interpreter.h | 46 size_t inputs_size() const { return subgraph_->inputs()->Length(); } in inputs_size() function
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 146 EXPECT_EQ(2, info.inputs_size()); in TEST()
148 EXPECT_EQ(3, info.inputs_size()); in TEST()
150 EXPECT_EQ(3, info.inputs_size()); in TEST()
175 EXPECT_EQ(0, info.inputs_size()); in TEST()
|
| D | op_level_cost_estimator.cc | 464 if (op_info.inputs_size() >= 2) { in PredictCwiseOp()
691 if (op_info.inputs_size() < 2) { in CountMatMulOperations()
692 LOG(ERROR) << "Need 2 inputs but got " << op_info.inputs_size(); in CountMatMulOperations()
765 if (op_info.inputs_size() != 2) { in CountBatchMatMulOperations()
766 LOG(ERROR) << "Expected 2 inputs but got " << op_info.inputs_size(); in CountBatchMatMulOperations()
926 if (op_info.inputs_size() < 2) { in CountConv2DBackpropInputOperations()
986 if (op_info.inputs_size() >= 2 && op_info.inputs(1).has_value()) { in CountConv2DBackpropFilterOperations()
1003 if (op_info.inputs_size() < 1) { in CountConv2DBackpropFilterOperations()
1339 if (op_info.outputs_size() == 0 || op_info.inputs_size() < inputs_needed) { in PredictGatherOrSlice()
1604 if (op_info.inputs_size() >= 1 && op_info.inputs(0).has_value()) { in PredictAvgPoolGrad()
|
| D | utils.cc | 103 if (i >= op_info->inputs_size()) { in ExtractExtraProperties()
111 if (input_node->op() == "Const" && i < op_info->inputs_size()) { in ExtractExtraProperties()
|
| D | virtual_scheduler_test.cc | 2385 EXPECT_EQ(2, ops_executed["c0"].op_info.inputs_size()); in TEST_F()
2386 EXPECT_EQ(2, ops_executed["c1"].op_info.inputs_size()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 177 const int inputs_size,
197 const int inputs_size, const std::vector<int>& extra_inputs,
|
| D | graph_transferer.cc | 855 const int padding, const int inputs_size, in AppendNodeParams() argument
864 node_info.set_input_count(inputs_size + in AppendNodeParams()
948 const int inputs_size, const std::vector<int>& extra_inputs, in AppendNodeParamsWithIoParams() argument
958 AppendNodeParams(name, id, type, type_id, padding, inputs_size, extra_inputs, in AppendNodeParamsWithIoParams()
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim_test.cc | 83 EXPECT_EQ(1, regression_signature_def.inputs_size()); in LoadAndValidateSavedModelBundle()
126 EXPECT_EQ(1, signature_def.inputs_size()); in TEST()
542 EXPECT_EQ(1, signature_def.inputs_size()); in TEST()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer.cc | 263 for (int j = 0; j < perf.op().inputs_size(); j++) { in PrintAnalysis()
275 if (j < perf.op().inputs_size() - 1) { in PrintAnalysis()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_interleave_dataset_op.cc | 822 int64 inputs_size; in ReadElement() local
825 &inputs_size)); in ReadElement()
826 element->inputs.resize(inputs_size); in ReadElement()
827 for (int i = 0; i < inputs_size; i++) { in ReadElement()
|