Home
last modified time | relevance | path

Searched defs:num_outputs (Results 1 – 25 of 99) sorted by relevance

1234

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py64 def run_core_tests(self, ds_fn1, ds_fn2, num_outputs, sparse_tensors=False): argument
112 num_outputs, argument
132 def verify_fully_used_iterator(self, ds_fn, num_outputs, argument
152 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
180 num_outputs, argument
204 num_outputs, argument
230 num_outputs, argument
284 num_outputs, argument
346 num_outputs, argument
407 num_outputs, argument
[all …]
Dserialization_integration_test.py31 def _build_input_pipeline(self, name, num_outputs): argument
40 def _build_graph(self, num_pipelines, num_outputs): argument
Dmap_dataset_serialization_test.py132 def _build_ds(num_outputs): argument
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column_ops.py273 num_outputs, argument
314 num_outputs, argument
362 num_outputs, argument
433 num_outputs, argument
Dlayers.py919 num_outputs, argument
1075 num_outputs, argument
1118 num_outputs, argument
1161 num_outputs, argument
1311 num_outputs, argument
1427 num_outputs, argument
1770 num_outputs, argument
2664 num_outputs, argument
/external/tensorflow/tensorflow/python/
Dpywrap_tfe.i275 int num_outputs = $1->size(); variable
/external/tensorflow/tensorflow/core/ops/
Dfunctional_ops_test.cc27 int num_outputs = 3; in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h167 int num_outputs() const override { return kernel_->num_outputs(); } in num_outputs() function
228 int num_outputs() const override { return output_dtypes_.size(); } in num_outputs() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h157 int num_outputs() const { return context_->num_outputs(); } in num_outputs() function
/external/tensorflow/tensorflow/python/eager/
Dexecute.py33 def quick_execute(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
75 def execute_with_callbacks(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
Dexecution_callbacks.py62 num_outputs, argument
117 def num_outputs(self): member in InfOrNanError
/external/libaom/libaom/av1/encoder/
Dml.h26 int num_outputs; // Number of output nodes. member
/external/tensorflow/tensorflow/python/ops/
Dsplit_benchmark.py58 def _run_graph(self, device, output_shape, variable, num_outputs, axis): argument
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dlayers.py97 num_outputs, argument
261 num_outputs, argument
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone_test.cc171 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST() local
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dinc_op_plugin.h58 const nvinfer1::Dims* outputs, int num_outputs, in configure()
/external/tensorflow/tensorflow/java/src/main/native/
Doperation_jni.cc95 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_Operation_shape() local
148 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_Operation_dtype() local
/external/tensorflow/tensorflow/contrib/slim/python/slim/nets/
Dresnet_utils.py87 def conv2d_same(inputs, num_outputs, kernel_size, stride, rate=1, scope=None): argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dshuffle_and_repeat_test.py37 def _gen_outputs(self, ds_fn, num_outputs, verify_exhausted=True): argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin.cc51 const nvinfer1::Dims* outputs, int num_outputs, in configure()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc245 const size_t num_outputs = kernel_state.num_outputs; in RunAsync() local
332 size_t num_outputs; member
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py65 def _DenseLayer(x, num_inputs, num_outputs, quantization_range, name): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc104 const int num_outputs = op_kernel->num_outputs(); in EvaluateNode() local
Dgraph_optimizer_stage.cc61 auto num_outputs = output_properties.size(); in GetTensorProperties() local
Dloop_optimizer.cc88 NodeDef* node, const int num_outputs) { in HandleInvariantEnter()
118 const int num_outputs, in HandleConst()
191 NodeDef* node, const int num_outputs, const int frame_id) { in HandleInvariantNode()
264 const int num_outputs = iter->second; in MoveInvariantNodes() local

1234