Home
last modified time | relevance | path

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

1234567891011

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dcheckpoint_test_base.py65 def verify_unused_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
75 def verify_fully_used_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
85 def verify_exhausted_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
95 def verify_multiple_breaks(obj, ds_fn, num_outputs, sparse_tensors=False): argument
107 num_outputs, argument
137 num_outputs, argument
160 num_outputs, argument
180 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
208 num_outputs, argument
235 num_outputs, argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A 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
H A Dnn_ops.cc35 int num_outputs = 1; in PYBIND11_MODULE() local
48 int num_outputs = 2; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A 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 …]
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dir.cpp64 Node::Node(OpKind op, size_t num_outputs) in Node()
71 size_t num_outputs) in Node()
96 size_t num_outputs) in Node()
101 Node::Node(OpKind op, OpList operands, size_t num_outputs) in Node()
104 Node::Node(OpKind op, Shape shape, size_t num_outputs) : Node(op, num_outputs) { in Node()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ops/
H A Dgeneric.cpp10 size_t num_outputs, in Generic()
19 size_t num_outputs, in Generic()
27 size_t num_outputs, in Generic()
31 Generic::Generic(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed) in Generic()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_node.cpp43 size_t num_outputs, in TsNode()
58 size_t num_outputs, in TsNode()
64 TsNode::TsNode(OpKind op, OpList operands, size_t num_outputs, hash_t hash_seed) in TsNode()
67 TsNode::TsNode(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed) in TsNode()
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dml_avx2.c31 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to1()
59 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to4()
89 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to8()
120 bool is_output_layer, int num_outputs, float *const output_nodes) { in nn_propagate_input_multiple_of_8()
153 const int num_outputs = is_output_layer in av1_nn_predict_avx2() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp643 const auto num_outputs = static_cast<uint32_t>(node->outputs().size()); in prepareBlockInfo() local
729 const auto num_outputs = static_cast<uint32_t>(node->inputs().size()); in prepareStaticNodeInfos() local
820 const auto num_outputs = static_cast<uint32_t>(outputs.size()); in prepare_for_memory_planner() local
860 size_t StaticModule::num_outputs() const { in num_outputs() function in torch::jit::StaticModule
1087 const auto num_outputs = static_cast<uint32_t>(p_node.num_outputs()); in destroyNodeOutputs() local
1124 c10::IValue BlockRunner::move_outputs_to_tuple(uint32_t num_outputs) { in move_outputs_to_tuple()
1657 const auto num_outputs = static_cast<uint32_t>(outputs.size()); in display_pnode_info() local
1856 const auto num_outputs = static_cast<uint32_t>(pnode.num_outputs()); in check_for_memory_leak() local
1936 const auto num_outputs = static_cast<uint32_t>(pnode.num_outputs()); in checkOutputTensorMemoryLeaks() local
1984 const auto num_outputs = static_cast<uint32_t>(n.outputs().size()); in disableManageOutputTensors() local
[all …]
H A Dimpl.h286 size_t num_outputs() const { in num_outputs() function
325 size_t num_outputs() const { in num_outputs() function
830 size_t num_outputs() const { in num_outputs() function
941 uint32_t num_outputs() const { in num_outputs() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DReduce.h120 …esult(const int index, const res_t result, const TensorIteratorBase &iter, const int num_outputs) { in set_result()
129 static void set_results(const res_t result, const TensorIteratorBase &iter, const int num_outputs) { in set_results()
151 …esults(const std::tuple<res_t...>& result, const TensorIteratorBase &iter, const int num_outputs) { in set_results()
212 const int num_outputs = iter.noutputs(); in binary_kernel_reduce() local
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DFullyConnectedLayer.cpp34 FullyConnectedLayerNode::FullyConnectedLayerNode(unsigned int num_outputs, QuantizationInfo out_qua… in FullyConnectedLayerNode()
56 … unsigned int num_outputs, in compute_weights_descriptor()
91 … unsigned int num_outputs, in compute_output_descriptor()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dexecute.py27 def quick_execute(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
72 num_outputs, argument
125 def execute_with_callbacks(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_function.cpp156 auto num_outputs = PyTuple_GET_SIZE(r.get()); in apply() local
436 auto num_outputs = PyTuple_GET_SIZE(outputs); in to_variable_list() local
612 Py_ssize_t num_outputs = PyTuple_GET_SIZE(raw_output); in _wrap_outputs() local
677 const int num_outputs = PyTuple_GET_SIZE(r.get()); in _wrap_outputs() local
1025 int num_outputs = PyTuple_GET_SIZE(output_objects); in _trace_post_record() local
1090 auto num_outputs = PyTuple_GET_SIZE(raw_output.get()); in process_outputs() local
1606 const auto num_outputs = cdata->num_outputs(); in THPFunction_next_functions() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute.cc1260 int num_outputs = kernel->num_outputs(); in GetOrCreateKernelAndDevice() local
1316 const int num_outputs = kernel->num_outputs(); in AddOrExecuteNode() local
1434 int num_outputs = kernel->num_outputs(); in EagerLocalExecute() local
1626 const size_t num_outputs = output_dtypes.size(); in EagerRemoteExecute() local
1699 std::vector<EagerKernelRet>* outputs, int num_outputs, in GetKernelOutputs()
2046 TensorHandle** retvals, int num_outputs, StatusCallback done) { in EagerKernelExecuteAsync()
2115 int num_outputs = kernel->num_outputs(); in EagerLocalExecuteAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dmake_saveable_from_iterator_test.py31 def _build_input_pipeline(self, name, num_outputs): argument
40 def _build_graph(self, num_pipelines, num_outputs): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc88 int num_outputs) { in PipeliningPartitionerRun()
642 TFE_TensorHandle** inputs, int* num_outputs, TFE_TensorHandle** outputs, in MaybeHandleDTensorCustomOps()
671 TFE_TensorHandle** outputs, int* num_outputs, in CopyToMesh()
1316 const int num_outputs, const ExecutionFunctions** execution_functions, in LowerToSPMDFunction()
1494 int* num_outputs, TFE_TensorHandle** outputs, TF_Status* status) { in ExecuteRegularOperation()
1829 void DTensorDevice::Execute(const TFE_Op* original_op, int* num_outputs, in Execute()
1962 void ExecuteOnDTensorDevice(const TFE_Op* original_op, int* num_outputs, in ExecuteOnDTensorDevice()
/aosp_15_r20/external/libaom/av1/encoder/
H A Dml.h26 int num_outputs; // Number of output nodes. member
41 const int num_outputs; // Number of output nodes. member
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod_meta.cpp147 size_t MethodMeta::num_outputs() const { in num_outputs() function in executorch::runtime::MethodMeta
152 auto num_outputs = this->num_outputs(); in output_tag() local
/aosp_15_r20/external/executorch/extension/pybindings/
H A Dpybindings.cpp555 size_t num_outputs() const { in num_outputs() function
753 const auto num_outputs = method.outputs_size(); in run_method() local
854 const auto num_outputs = method.outputs_size(); in plan_execute() local
927 const auto num_outputs = method.outputs_size(); in make_output_storages() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dutils.cc152 int num_outputs = 0; in NumOutputs() local
244 int num_outputs = 0; in NumControlOutputs() local
260 int num_outputs = 0; in NumNonControlOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dcostmodel.cc128 void CostModel::Ensure(int id, int num_outputs) { in Ensure()
156 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs()
418 const int num_outputs = n->num_outputs(); in AddNodesToCostModel() local
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A 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
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dinterpreter.cpp479 auto num_outputs = frames.back().function->n_outputs; in runTemplate() local
783 auto num_outputs = fn.graph()->outputs().size(); in runTemplate() local
1103 const auto num_outputs = frames.front().function->n_outputs; in run() local
1187 size_t Code::num_outputs() const { in num_outputs() function in torch::jit::Code

1234567891011