| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm.py | 43 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
127 def RandomInputs(batch_size, seq_length, num_inputs): argument
140 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
|
| /external/libaom/libaom/av1/encoder/arm/neon/ |
| D | ml_neon.c | 31 static void nn_propagate_8to1(int num_inputs, const float *const inputs, in nn_propagate_8to1()
61 static void nn_propagate_xto1(int num_inputs, const float *const inputs, in nn_propagate_xto1()
96 static void nn_propagate_xsto1(int num_inputs, const float *const inputs, in nn_propagate_xsto1()
115 static void nn_propagate_4to1(int num_inputs, const float *const inputs, in nn_propagate_4to1()
141 static void nn_propagate_4to4(int num_inputs, const float *const inputs, in nn_propagate_4to4()
179 static void nn_propagate_4to8(const int num_inputs, const float *const inputs, in nn_propagate_4to8()
224 static void nn_propagate_8to4(const int num_inputs, const float *const inputs, in nn_propagate_8to4()
271 int num_inputs = nn_config->num_inputs; in av1_nn_predict_neon() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_dataset_op.cc | 55 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
68 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
96 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
205 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize() local
318 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in StartThreads() local
350 for (size_t i = 1, num_inputs = histograms_.size(); i < num_inputs; in SelectFastestInputIndex() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_aggregate_ops.cc | 51 const int num_inputs = ctx->num_inputs() / 2; in CheckInputShape() local
73 const int num_inputs = ctx->num_inputs() / 2; in FindMKLInputIndex() local
88 const int num_inputs = ctx->num_inputs() / 2; in ComputeScalar() local
114 const int num_inputs = ctx->num_inputs() / 2; in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | concat_benchmark.py | 35 def build_graph(device, input_shape, variable, num_inputs, axis, grad): argument
81 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add_n.h | 28 inline void AddN(const RuntimeShape& input_shape, const size_t num_inputs, in AddN()
43 const RuntimeShape& input_shape, const size_t num_inputs, in AddN()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 32 int num_inputs = NumInputs(node); in Prepare() local
65 int num_inputs = NumInputs(node); in EvalAddN() local
|
| D | add_n_test.cc | 33 int num_inputs = inputs.size(); in BaseAddNOpModel() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 68 int num_inputs = branch_bodies[0]->fdef.signature().input_arg_size(); in CondConstInputIndices() local
102 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs() local
154 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs() local
|
| D | xla_op_kernel.h | 77 int num_inputs() const { return context_->num_inputs(); } in num_inputs() function
|
| D | mlir_xla_op_kernel.cc | 28 int num_inputs = ctx->num_inputs(); in ContextToXlaArgs() local
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 63 const int num_inputs) { in nn_propagate_4to4()
85 __m128 *const out_l, const int num_inputs) { in nn_propagate_4to8()
112 const int num_inputs) { in nn_propagate_8to4()
158 int num_inputs = nn_config->num_inputs; in av1_nn_predict_sse3() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | ml.h | 25 int num_inputs; // Number of input nodes, i.e. features. member
40 const int num_inputs; // Number of input nodes, i.e. features. member
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | signature_def_function.cc | 40 int num_inputs, TF_Status* status) { in TF_SignatureDefFunctionMakeCallOp()
|
| D | concrete_function.cc | 38 TFE_TensorHandle** inputs, int num_inputs, in TF_ConcreteFunctionMakeCallOp()
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 92 int num_inputs() { return n_input_; } in num_inputs() function in tflite::ops::custom::__anone91f80820111::MicroFrontendOpModel
122 const int num_inputs = micro_frontend->num_inputs(); in VerifyGoldens() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLConcatenateLayer.cpp | 150 const unsigned int num_inputs = inputs_vector.size(); in validate() local
260 unsigned int num_inputs{ 0 }; member
|
| D | CLStackLayer.cpp | 78 const unsigned int num_inputs = input.size(); in validate() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_format_op_test.cc | 32 Status Init(int num_inputs, DataType input_type, in Init()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEStackLayer.cpp | 71 const unsigned int num_inputs = input.size(); in validate() local
|
| /external/mesa3d/src/gallium/drivers/svga/ |
| D | svga_link.h | 11 unsigned num_inputs; /* number of inputs in the current shader */ member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops_test.cc | 56 int num_inputs = 4; in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 192 const int num_inputs = node.input_size(); in HasControlInputs() local
200 const int num_inputs = node.input_size(); in HasRegularInputs() local
208 int num_inputs = 0; in NumNonControlInputs() local
219 int num_inputs = 0; in NumControlInputs() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 141 int num_inputs = input_str.size(); in ParseEquation() local
229 const int num_inputs = inputs.size(); in ProcessDimensions() local
623 const int num_inputs = inputs.size(); in Compute() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 161 const int num_inputs = node.input_size(); in NumEdgesIfBypassed() local
200 const int num_inputs = node.input_size(); in BypassingNodeIsBeneficial() local
382 const int num_inputs = node->input_size(); in OptimizeNode() local
|