Searched defs:inputs_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter.h | 46 size_t inputs_size() const { return subgraph_->inputs()->Length(); } in inputs_size() function
|
| D | micro_interpreter_test.cc | 123 constexpr size_t inputs_size = 1; in BuildMockModel() local
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax_test.cc | 36 constexpr int inputs_size = 2; in TestSoftmaxFloat() local
100 constexpr int inputs_size = 1; in TestSoftmaxQuantized() local
|
| D | depthwise_conv_test.cc | 43 constexpr int inputs_size = 3; in TestDepthwiseConvFloat() local
124 constexpr int inputs_size = 3; in TestDepthwiseConvQuantized() local
|
| D | fully_connected_test.cc | 43 constexpr int inputs_size = 3; in TestFullyConnectedFloat() local
117 constexpr int inputs_size = 3; in TestFullyConnectedQuantized() local
|
| /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
112 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
186 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 89 const size_t inputs_size = PyList_Size(input_value); in FeedTensor() local
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 855 const int padding, const int inputs_size, in AppendNodeParams()
948 const int inputs_size, const std::vector<int>& extra_inputs, in AppendNodeParamsWithIoParams()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_interleave_dataset_op.cc | 822 int64 inputs_size; in ReadElement() local
|