| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm.py | 40 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
124 def RandomInputs(batch_size, seq_length, num_inputs): argument
137 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
|
| /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/core/kernels/data/experimental/ |
| D | choose_fastest_dataset_op.cc | 54 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
67 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
95 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
189 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize() local
300 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in StartThreads() local
331 for (size_t i = 1, num_inputs = histograms_.size(); i < num_inputs; in SelectFastestInputIndex() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 30 int num_inputs = NumInputs(node); in Prepare() local
57 int num_inputs = NumInputs(node); in EvalAddN() local
|
| D | concatenation_test.cc | 33 int num_inputs) { in BaseConcatenationOpModel()
65 int axis, int num_inputs, in QuantizedConcatenationOpModel()
|
| D | add_n_test.cc | 32 int num_inputs = inputs.size(); in BaseAddNOpModel() local
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_atom_array.c | 320 unsigned num_inputs) in is_interleaved_arrays()
458 unsigned num_inputs) in setup_interleaved_attribs()
577 unsigned num_inputs) in setup_non_interleaved_attribs()
691 unsigned num_inputs; in st_update_array() local
|
| /external/mesa3d/src/gallium/drivers/svga/ |
| D | svga_link.h | 11 unsigned num_inputs; member
|
| /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/tensorflow/tensorflow/core/ops/ |
| D | functional_ops_test.cc | 26 int num_inputs = 4; in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 166 int num_inputs() const override { return kernel_->num_inputs(); } in num_inputs() function
227 int num_inputs() const override { return input_dtypes_.size(); } in num_inputs() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 71 int num_inputs() const { return context_->num_inputs(); } in num_inputs() function
|
| D | const_analysis.cc | 152 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | ml.h | 25 int num_inputs; // Number of input nodes, i.e. features. member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.h | 38 size_t num_inputs = inputs.size(); in ConcatCPUImpl() local
140 size_t num_inputs = inputs.size(); in ConcatSYCLImpl() local
|
| D | string_format_op_test.cc | 31 Status Init(int num_inputs, DataType input_type, in Init()
|
| D | tridiagonal_solve_op.cc | 40 auto num_inputs = input_matrix_shapes.size(); in ValidateInputMatrixShapes() local
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 93 int num_inputs() { return n_input_; } in num_inputs() function in tflite::ops::custom::__anond16ec6df0111::MicroFrontendOpModel
123 const int num_inputs = micro_frontend->num_inputs(); in VerifyGoldens() local
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_plugin.h | 57 void configure(const nvinfer1::Dims* inputs, int num_inputs, in configure()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.cc | 50 void PluginTensorRT::configure(const nvinfer1::Dims* inputs, int num_inputs, in configure()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 244 const size_t num_inputs = kernel_state.num_inputs; in RunAsync() local
330 size_t num_inputs; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 133 const int num_inputs = node.input_size(); in NumEdgesIfBypassed() local
172 const int num_inputs = node.input_size(); in BypassingNodeIsBeneficial() local
356 const int num_inputs = node->input_size(); in OptimizeNode() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | remove_nodes.cc | 55 for (int num_inputs = 1; num_inputs <= max_inputs; ++num_inputs) { in RemoveNodes() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | benchmark_testlib.h | 123 const uint32 num_inputs = rnd.Uniform(std::min(i, 5)); in CreateRandomGraph() local
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_mnist_test.py | 65 def _DenseLayer(x, num_inputs, num_outputs, quantization_range, name): argument
|