Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.cc35 int num_outputs = 1; in PYBIND11_MODULE() local
45 int num_outputs = 1; in PYBIND11_MODULE() local
57 int num_outputs = 1; in PYBIND11_MODULE() local
67 int num_outputs = 1; in PYBIND11_MODULE() local
77 int num_outputs = 1; in PYBIND11_MODULE() local
87 int num_outputs = 1; in PYBIND11_MODULE() local
97 int num_outputs = 1; in PYBIND11_MODULE() local
Dnn_ops.cc35 int num_outputs = 1; in PYBIND11_MODULE() local
48 int num_outputs = 2; in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test_base.py70 def verify_unused_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
80 def verify_fully_used_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
90 def verify_exhausted_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
100 def verify_multiple_breaks(obj, ds_fn, num_outputs, sparse_tensors=False): argument
112 num_outputs, argument
142 num_outputs, argument
165 num_outputs, argument
185 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
213 num_outputs, argument
240 num_outputs, argument
[all …]
Dtake_test.py60 def test(self, verify_fn, count, num_outputs): argument
Dskip_test.py61 def test(self, verify_fn, count, num_outputs): argument
/external/ComputeLibrary/src/graph/nodes/
DFullyConnectedLayer.cpp34 FullyConnectedLayerNode::FullyConnectedLayerNode(unsigned int num_outputs, QuantizationInfo out_qua… in FullyConnectedLayerNode()
47 … unsigned int num_outputs, in compute_weights_descriptor()
82 … unsigned int num_outputs, in compute_output_descriptor()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmake_saveable_from_iterator_test.py35 def _build_input_pipeline(self, name, num_outputs): argument
44 def _build_graph(self, num_pipelines, num_outputs): argument
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops_test.cc79 const int num_outputs = 4; in TEST() local
117 const int num_outputs = 4; in TEST() local
135 const int num_outputs, Graph* graph, in CreateSplitTensorGraph()
162 const int num_outputs, Graph* graph, in CreateSplitResourceGraph()
217 const int num_outputs = 1; in TEST_P() local
235 const int num_outputs = 1; in TEST_P() local
253 const int num_outputs = 1; in TEST_P() local
271 const int num_outputs = 4; in TEST_P() local
288 const int num_outputs = 4; in TEST_P() local
306 const int num_outputs = 2; in TEST_P() local
[all …]
/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/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/core/protobuf/
Ddebug_event.proto185 int32 num_outputs = 7; field
239 int32 num_outputs = 2; field
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_link.h17 unsigned num_outputs; member
/external/tensorflow/tensorflow/python/ops/
Dsplit_benchmark.py58 def _run_graph(self, device, output_shape, variable, num_outputs, axis): argument
/external/mesa3d/src/freedreno/vulkan/
Dtu_nir_lower_multiview.c89 unsigned num_outputs; in tu_nir_lower_multiview() local
/external/tensorflow/tensorflow/core/ops/
Dfunctional_ops_test.cc57 int num_outputs = 3; in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h217 int num_outputs() const { return context_->num_outputs(); } in num_outputs() function
/external/tensorflow/tensorflow/core/data/
Dstandalone_test.cc290 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST() 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/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/core/common_runtime/eager/
Dexecute.cc990 int num_outputs = kernel->num_outputs(); in GetOrCreateKernelAndDevice() local
1046 const int num_outputs = kernel->num_outputs(); in AddOrExecuteNode() local
1155 int num_outputs = kernel->num_outputs(); in EagerLocalExecute() local
1345 const size_t num_outputs = output_dtypes.size(); in EagerRemoteExecute() local
1415 std::vector<EagerKernelRet>* outputs, int num_outputs, in GetKernelOutputs()
1751 TensorHandle** retvals, int num_outputs, StatusCallback done) { in EagerKernelExecuteAsync()
1820 int num_outputs = kernel->num_outputs(); in EagerLocalExecuteAsync() local
Dkernel_and_device.h243 int num_outputs() const override { return kernel_->num_outputs(); } in num_outputs() function
329 int num_outputs() const override { return output_dtypes_.size(); } in num_outputs() function
/external/ComputeLibrary/tests/validation/fixtures/
DLSTMLayerFixture.h78 const unsigned int num_outputs = recurrent_weights_shape.x(); in compute_target() local
323 const unsigned int num_outputs = recurrent_weights_shape.x(); in compute_reference() 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

1234567