Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc67 TfLiteTensor* output_values; in ResizeOutput() local
177 int32* output_indexes, T* output_values) { in TopK()
206 TfLiteTensor* output_values; in Prepare() local
222 TfLiteTensor* output_values; in Prepare() local
232 TfLiteTensor* output_values; in Eval() local
Dunique.cc78 std::vector<T> output_values; in EvalImpl() local
Dmaximum_minimum_test.cc79 std::initializer_list<data_type> output_values, in TestModel()
/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_softmax_op.cc74 Tensor *output_values = nullptr; in Compute() local
Dsparse_dense_binary_op_shared.cc104 Tensor *output_values = nullptr; in Compute() local
Dserialize_sparse_op.cc194 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in operator ()() local
294 Tensor& output_values = serialized_sparse_t(b, 1).emplace<Tensor>( in operator ()() local
Dsparse_tensors_map_ops.cc297 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
389 const Tensor& output_values = st.values(); in Compute() local
Dsparse_fill_empty_rows_op.cc148 auto output_values = output_values_t->vec<T>(); in operator ()() local
Dcount_ops.cc50 auto output_values = values->flat<T>(); in OutputSparse() local
Dquantization_utils_test.cc50 auto output_values = o_tensor.flat<quint8>(); in TestRequantizeMany() local
92 auto output_values = o_tensor.flat<qint32>(); in TestRequantizeMany8To32Bit() 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.h487 for (float& output_values : output_ref) { variable
554 for (float& output_values : output_ref) { variable
610 for (float& output_values : output_ref) { variable
690 for (float& output_values : output_ref) { variable
/external/tensorflow/tensorflow/lite/micro/kernels/
Dhard_swish_test.cc264 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
280 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
307 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
335 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
Dconcatenation_test.cc198 const float output_values[] = {1.0f, 3.0f, 1.0f, 2.0f, 3.0f, 4.0f, in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc329 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon36a8fc880e02() local
372 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon36a8fc881002() local
624 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon36a8fc881802() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dtftext_utils.cc164 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
182 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc178 Tensor output_values(cpu_allocator(), DataTypeToEnum<T>::value, in Compute() local
Dsparse_mat_mul_op.cc218 Tensor output_values(cpu_allocator(), DataTypeToEnum<T>::value, 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
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h499 std::vector<Tensor> output_values; in Split() local

12