Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.cc35 int num_outputs = 1; in PYBIND11_MODULE() local
46 int num_outputs = 1; in PYBIND11_MODULE() local
58 int num_outputs = 1; in PYBIND11_MODULE() local
69 int num_outputs = 1; in PYBIND11_MODULE() local
80 int num_outputs = 1; in PYBIND11_MODULE() local
91 int num_outputs = 1; in PYBIND11_MODULE() local
102 int num_outputs = 1; in PYBIND11_MODULE() local
Dnn_ops.cc35 int num_outputs = 1; in PYBIND11_MODULE() local
49 int num_outputs = 2; in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py64 def run_core_tests(self, ds_fn, num_outputs, sparse_tensors=False): argument
96 num_outputs, argument
116 def verify_fully_used_iterator(self, ds_fn, num_outputs, argument
136 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
164 num_outputs, argument
190 num_outputs, argument
244 num_outputs, argument
276 num_outputs, argument
316 num_outputs, argument
425 def gen_break_points(self, num_outputs, num_samples=10): argument
Dserialization_integration_test.py35 def _build_input_pipeline(self, name, num_outputs): argument
44 def _build_graph(self, num_pipelines, num_outputs): argument
/external/tensorflow/tensorflow/python/eager/
Dexecute.py33 def quick_execute(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
81 num_outputs, argument
136 def execute_with_callbacks(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
/external/libaom/libaom/av1/encoder/
Dml.h26 int num_outputs; // Number of output nodes. member
41 const 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/mesa3d/src/gallium/drivers/svga/
Dsvga_link.h17 unsigned num_outputs; member
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc561 int num_outputs = kernel->num_outputs(); in GetOrCreateKernelAndDevice() local
617 const int num_outputs = kernel->num_outputs(); in AddOrExecuteNode() local
726 int num_outputs = kernel->num_outputs(); in EagerLocalExecute() local
918 const size_t num_outputs = output_dtypes.size(); in EagerRemoteExecute() local
988 std::vector<EagerKernelRet>* outputs, int num_outputs, in GetKernelOutputs()
1319 TensorHandle** retvals, int num_outputs, StatusCallback done) { in EagerKernelExecuteAsync()
1383 int num_outputs = kernel->num_outputs(); in EagerLocalExecuteAsync() local
Dkernel_and_device.h239 int num_outputs() const override { return kernel_->num_outputs(); } in num_outputs() function
322 int num_outputs() const override { return output_dtypes_.size(); } in num_outputs() function
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_operation_jni.cc89 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_GraphOperation_shape() local
139 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_GraphOperation_dtype() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h172 int num_outputs() const { return context_->num_outputs(); } in num_outputs() function
/external/mesa3d/src/freedreno/vulkan/
Dtu_nir_lower_multiview.c89 unsigned num_outputs; in tu_nir_lower_multiview() local
/external/tensorflow/tensorflow/core/protobuf/
Ddebug_event.proto185 int32 num_outputs = 7; field
239 int32 num_outputs = 2; field
/external/tensorflow/tensorflow/core/ops/
Dfunctional_ops_test.cc57 int num_outputs = 3; in TEST() local
/external/tensorflow/tensorflow/core/data/
Dstandalone_test.cc290 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc83 const int num_outputs = op_kernel->num_outputs(); in EvaluateNode() local
Dloop_optimizer.cc87 NodeDef* node, const int num_outputs) { in HandleInvariantEnter()
117 const int num_outputs, in HandleConst()
189 NodeDef* node, const int num_outputs, const int frame_id) { in HandleInvariantNode()
262 const int num_outputs = iter->second; in MoveInvariantNodes() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental.cc177 void TF_OutputListSetNumOutputs(TF_OutputList* o, int num_outputs, in TF_OutputListSetNumOutputs()
229 int num_outputs = unwrap(o)->expected_num_outputs; in TF_ExecuteOperation() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dshuffle_and_repeat_test.py38 def _gen_outputs(self, ds_fn, num_outputs, verify_exhausted=True): argument
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h250 int num_outputs = 1; in ScalarsFromSequence() local
/external/mesa3d/src/mesa/state_tracker/
Dst_program.c437 unsigned num_outputs = 0; in st_prepare_vertex_program() local
454 unsigned num_outputs = 0; in st_translate_stream_output_info() local
492 unsigned num_outputs = 0; in st_translate_vertex_program() local
1587 uint num_outputs = 0; in st_translate_common_program() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc88 const int num_outputs = n->num_outputs(); in NodeItemBytes() local
145 const int num_outputs = n->num_outputs(); in InitializeNode() local
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py66 def _DenseLayer(x, num_inputs, num_outputs, quantization_range, name): argument
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc127 void CostModel::Ensure(int id, int num_outputs) { in Ensure()
155 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs()
417 const int num_outputs = n->num_outputs(); in AddNodesToCostModel() local

123456