Home
last modified time | relevance | path

Searched defs:output_values (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc67 TfLiteTensor* output_values; in ResizeOutput() local
178 int32* output_indexes, T* output_values) { in TopK()
207 TfLiteTensor* output_values; in Prepare() local
224 TfLiteTensor* output_values; in Prepare() local
234 TfLiteTensor* output_values; in Eval() local
Dunique.cc78 std::vector<T> output_values; in EvalImpl() local
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference_beta.cc168 T* __restrict output_values) { in ConcatV2Args2()
189 T* __restrict output_values) { in MatMul()
225 T* __restrict output_values) { in DepthwiseConv2dNative()
314 T* __restrict output_values) { in DepthwiseConv2dNativeVALID()
329 T* __restrict output_values) { in DepthwiseConv2dNativeSAME()
343 T* __restrict output_values) { in FullyConnected()
377 T* __restrict output_values) { in SpaceToBatchNDRank4()
448 T* __restrict output_values) { in BatchToSpaceNDRank4()
512 T* __restrict output_values) { in SparseDenseMatMulCSR()
533 T* __restrict output_values) { in SparseFullyConnectedCSR()
[all …]
Donedevice_train_palm_detection_filter_inference.cc169 T* __restrict output_values) { in ConcatV2Args2()
190 T* __restrict output_values) { in MatMul()
226 T* __restrict output_values) { in DepthwiseConv2dNative()
315 T* __restrict output_values) { in DepthwiseConv2dNativeVALID()
330 T* __restrict output_values) { in DepthwiseConv2dNativeSAME()
344 T* __restrict output_values) { in FullyConnected()
378 T* __restrict output_values) { in SpaceToBatchNDRank4()
449 T* __restrict output_values) { in BatchToSpaceNDRank4()
513 T* __restrict output_values) { in SparseDenseMatMulCSR()
534 T* __restrict output_values) { in SparseFullyConnectedCSR()
[all …]
Donedevice_train_palm_detection_filter_inference_v2.cc168 T* __restrict output_values) { in ConcatV2Args2()
189 T* __restrict output_values) { in MatMul()
225 T* __restrict output_values) { in DepthwiseConv2dNative()
314 T* __restrict output_values) { in DepthwiseConv2dNativeVALID()
329 T* __restrict output_values) { in DepthwiseConv2dNativeSAME()
343 T* __restrict output_values) { in FullyConnected()
377 T* __restrict output_values) { in SpaceToBatchNDRank4()
448 T* __restrict output_values) { in BatchToSpaceNDRank4()
512 T* __restrict output_values) { in SparseDenseMatMulCSR()
533 T* __restrict output_values) { in SparseFullyConnectedCSR()
[all …]
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dwhitespace_tokenizer.cc94 const TfLiteTensor* input, TfLiteTensor* output_values) { in WritePaddedOutput()
121 const TfLiteTensor* input, TfLiteTensor* output_values, in WriteRaggedOutput()
153 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
195 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Eval() local
Dngrams.cc90 TfLiteTensor* output_values = GetOutput(context, node, kValues); in Prepare() local
188 TfLiteTensor* output_values = GetOutput(context, node, kValues); in Eval() local
/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_string_op.cc83 Tensor output_values; in Compute() local
106 Tensor output_values; in Compute() local
232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
Ddeserialize_sparse_variant_op.cc60 const Tensor* output_values; in Compute() local
153 Tensor* output_values; in Compute() local
305 const Tensor** output_values) { in GetAndValidateSparseTensorIndicesAndValues()
Dsparse_slice_op_gpu.cu.cc94 int64_t* __restrict__ output_indices, T* __restrict__ output_values) { in SparseSliceGatherKernel()
156 Tensor* output_values = nullptr; in operator ()() local
223 Tensor* output_values = nullptr; in operator ()() local
Dsparse_fill_empty_rows_op_gpu.cu.cc132 const Tindex* num_new_rows_before, Tindex* output_indices, T* output_values, in ScatterInputElementsKernel()
154 const bool* empty_row_indicator, Tindex* output_indices, T* output_values) { in ScatterNewElementsKernel()
191 T* output_values; in operator ()() local
375 T* output_values; in operator ()() local
Dsparse_softmax_op.cc77 Tensor *output_values = nullptr; in Compute() local
Dsparse_dense_binary_op_shared.cc124 Tensor *output_values = nullptr; in Compute() local
Dserialize_sparse_op.cc196 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in operator ()() local
298 Tensor& output_values = serialized_sparse_t(b, 1).emplace<Tensor>( in operator ()() local
Dsparse_split_op_gpu.cu.cc170 T* __restrict__* __restrict__ output_values = in SparseSplitScatterKernel() local
314 GpuDeviceArrayOnHost<T*> output_values(context, num_split); in operator ()() local
Dsparse_tensors_map_ops.cc308 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
400 const Tensor& output_values = st.values(); in Compute() local
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_detokenizer_tflite.cc46 TfLiteTensor& output_values = in Prepare() local
83 TfLiteTensor& output_values = in Eval() local
Dsentencepiece_tokenizer_tflite.cc58 TfLiteTensor& output_values = in Prepare() local
98 TfLiteTensor& output_values = in Eval() local
/external/XNNPACK/test/
Dgavgpool-microkernel-tester.h446 for (float& output_values : output_ref) { in Test() local
514 for (float& output_values : output_ref) { in Test() local
570 for (float& output_values : output_ref) { in Test() local
632 for (float& output_values : output_ref) { in Test() local
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc333 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anonc55240e60e02() local
376 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anonc55240e61002() local
630 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anonc55240e61802() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dtftext_utils.cc163 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
181 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_session_coordination_test.cc135 TF_Tensor* output_values[1] = {nullptr}; in TEST_P() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc180 Tensor output_values(cpu_allocator(), DataTypeToEnum<T>::value, in Compute() local
/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
Dmultiplex_3_kernel.cc153 auto output_values = output_values_tensor->flat<T>(); in Compute() local
/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run() local

12