| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 67 TfLiteTensor* output_values; in ResizeOutput() local
69 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in ResizeOutput()
72 output_values->type = input->type; in ResizeOutput()
86 resize_tensor(output_values, output_values_shape, nullptr)); in ResizeOutput()
177 int32* output_indexes, T* output_values) { in TopK() argument
188 T* output_row = output_values + row * k; in TopK()
206 TfLiteTensor* output_values; in Prepare() local
208 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in Prepare()
209 TF_LITE_ENSURE_TYPES_EQ(context, input->type, output_values->type); in Prepare()
222 TfLiteTensor* output_values; in Prepare() local
[all …]
|
| D | unique.cc | 78 std::vector<T> output_values; in EvalImpl() local
91 output_values.push_back(data[i]); in EvalImpl()
104 for (int i = 0; i < output_values.size(); ++i) { in EvalImpl()
105 output_unique_values[i] = output_values[i]; in EvalImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_variant_op.cc | 60 const Tensor* output_values; in Compute() local
70 &output_values)); in Compute()
72 context->set_output(1, *output_values); in Compute()
153 Tensor* output_values; in Compute() local
162 1, {static_cast<int64>(total_non_zeros)}, &output_values)); in Compute()
259 output_values->vec<T>().slice(values_start, values_sizes) = \ in Compute()
305 const Tensor** output_values) { in GetAndValidateSparseTensorIndicesAndValues() argument
334 *output_values = serialized_values.get<Tensor>(); in GetAndValidateSparseTensorIndicesAndValues()
335 if (*output_values == nullptr) { in GetAndValidateSparseTensorIndicesAndValues()
339 if (!TensorShapeUtils::IsVector((*output_values)->shape())) { in GetAndValidateSparseTensorIndicesAndValues()
[all …]
|
| D | deserialize_sparse_string_op.cc | 83 Tensor output_values; in Compute() local
89 &output_indices, &output_values, &output_shape)); in Compute()
91 context->set_output(1, output_values); in Compute()
106 Tensor output_values; in Compute() local
112 &output_values, &output_shape)); in Compute()
137 values.push_back(output_values); in Compute()
232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor() argument
245 TF_RETURN_IF_ERROR(this->Deserialize(serialized_values, output_values)); in GetAndValidateSparseTensor()
246 if (!TensorShapeUtils::IsVector(output_values->shape())) { in GetAndValidateSparseTensor()
250 output_values->shape().DebugString()); in GetAndValidateSparseTensor()
[all …]
|
| D | sparse_tensors_map_ops.cc | 297 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
300 auto output_values_t = output_values.vec<T>(); in Compute()
311 SparseTensor::Create(output_indices, output_values, in Compute()
389 const Tensor& output_values = st.values(); in Compute() local
397 OP_REQUIRES(context, TensorShapeUtils::IsVector(output_values.shape()), in Compute()
401 output_values.shape().DebugString())); in Compute()
403 context, DataTypeToEnum<T>::value == output_values.dtype(), in Compute()
407 "].values.dtype() == ", DataTypeString(output_values.dtype()))); in Compute()
410 OP_REQUIRES(context, num_entries == output_values.dim_size(0), in Compute()
415 " vs. ", output_values.dim_size(0))); in Compute()
[all …]
|
| D | sparse_softmax_op.cc | 74 Tensor *output_values = nullptr; in Compute() local
76 &output_values)); in Compute()
77 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 104 Tensor *output_values = nullptr; in Compute() local
108 ctx->allocate_output(0, TensorShape({nnz}), &output_values)); in Compute()
159 output_values->flat<T>().device(ctx->eigen_device<Device>()) = in Compute()
|
| D | serialize_sparse_op.cc | 194 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in operator ()() local
197 auto output_values_t = output_values.vec<T>(); in operator ()()
207 serialize(output_values, &serialized_sparse_t(b, 1)); in operator ()()
294 Tensor& output_values = serialized_sparse_t(b, 1).emplace<Tensor>( in operator ()() local
301 T* output_values_ptr = static_cast<T*>(DMAHelper::base(&output_values)); in operator ()()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | whitespace_tokenizer.cc | 94 const TfLiteTensor* input, TfLiteTensor* output_values) { in WritePaddedOutput() argument
115 buffer.WriteToTensor(output_values, output_shape); in WritePaddedOutput()
121 const TfLiteTensor* input, TfLiteTensor* output_values, in WriteRaggedOutput() argument
148 buffer.WriteToTensor(output_values, output_shape); in WriteRaggedOutput()
153 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
154 SetTensorToDynamic(output_values); in Prepare()
195 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Eval() local
196 TF_LITE_ENSURE(context, IsDynamicTensor(output_values)); in Eval()
199 return WritePaddedOutput(list_of_tokens, input, output_values); in Eval()
209 return WriteRaggedOutput(list_of_tokens, input, output_values, in Eval()
|
| D | ngrams.cc | 90 TfLiteTensor* output_values = GetOutput(context, node, kValues); in Prepare() local
101 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_values, in Prepare()
106 SetTensorToDynamic(output_values); in Prepare()
188 TfLiteTensor* output_values = GetOutput(context, node, kValues); in Eval() local
190 buffer.WriteToTensor(output_values, /*new_shape=*/nullptr); in Eval()
192 buffer.WriteToTensorAsVector(output_values); in Eval()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_detokenizer_tflite.cc | 46 TfLiteTensor& output_values = in Prepare() local
48 SetTensorToDynamic(&output_values); in Prepare()
83 TfLiteTensor& output_values = in Eval() local
85 buf.WriteToTensor(&output_values, nullptr); in Eval()
|
| D | sentencepiece_tokenizer_tflite.cc | 58 TfLiteTensor& output_values = in Prepare() local
60 SetTensorToDynamic(&output_values); in Prepare()
98 TfLiteTensor& output_values = in Eval() local
102 context, &output_values, in Eval()
104 int32_t* output_values_flat = output_values.data.i32; in Eval()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 164 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
165 if (!output_values || !output_values.getElementType().isa<StringType>()) { in VerifyNgrams()
170 if (input_values.hasRank() && output_values.hasRank() && in VerifyNgrams()
171 input_values.getRank() != output_values.getRank()) { in VerifyNgrams()
182 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
183 if (!RankEquals(output_values, 1) || in VerifyNgrams()
184 !output_values.getElementType().isa<StringType>()) { in VerifyNgrams()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | hard_swish_test.cc | 264 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
267 input_values, output_values); in TF_LITE_MICRO_TEST()
280 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
292 input_values, output_values); in TF_LITE_MICRO_TEST()
307 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
319 input_values, output_values); in TF_LITE_MICRO_TEST()
335 float output_values[size] = {0.f}; in TF_LITE_MICRO_TEST() local
339 25.036512f, -0.3905796f, 24.50887f, 0.035, input_values, output_values); in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseReduceMaxSparse.pbtxt | 24 name: "output_values"
79 name: "output_values"
136 name: "output_values"
194 name: "output_values"
|
| D | SparseSparseMaximum.pbtxt | 32 name: "output_values"
84 name: "output_values"
138 name: "output_values"
193 name: "output_values"
|
| D | SparseReduceSumSparse.pbtxt | 24 name: "output_values"
84 name: "output_values"
146 name: "output_values"
209 name: "output_values"
|
| D | SparseSparseMinimum.pbtxt | 32 name: "output_values"
89 name: "output_values"
148 name: "output_values"
208 name: "output_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseSparseMaximum.pbtxt | 32 name: "output_values"
84 name: "output_values"
138 name: "output_values"
193 name: "output_values"
|
| D | SparseReduceMaxSparse.pbtxt | 24 name: "output_values"
79 name: "output_values"
136 name: "output_values"
194 name: "output_values"
|
| D | SparseReduceSumSparse.pbtxt | 24 name: "output_values"
84 name: "output_values"
146 name: "output_values"
209 name: "output_values"
|
| D | SparseSparseMinimum.pbtxt | 32 name: "output_values"
89 name: "output_values"
148 name: "output_values"
208 name: "output_values"
|
| /external/XNNPACK/test/ |
| D | gavgpool-microkernel-tester.h | 487 for (float& output_values : output_ref) { variable
488 output_values = std::max(std::min(output_values, output_max), output_min);
554 for (float& output_values : output_ref) { variable
555 output_values = std::max(std::min(output_values, output_max), output_min);
610 for (float& output_values : output_ref) { variable
611 output_values = std::max(std::min(output_values, output_max), output_min);
690 for (float& output_values : output_ref) { variable
691 output_values = std::max(std::min(output_values, output_max), output_min);
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 499 std::vector<Tensor> output_values; in Split() local
502 output_values.reserve(num_split); in Split()
537 output_values.emplace_back(DataTypeToEnum<T>::v(), in Split()
541 output_values_t.emplace_back(output_values[i].vec<T>()); in Split()
566 Create(output_indices[i], output_values[i], output_shapes[i], &tensor); in Split()
628 Tensor output_values(DataTypeToEnum<T>::v(), TensorShape({count})); in Slice()
631 auto output_values_t = output_values.vec<T>(); in Slice()
659 return SparseTensor(output_indices, output_values, output_shape); in Slice()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | session_jni.cc | 148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run() local
177 static_cast<int>(ninputs), outputs.get(), output_values.get(), in Java_org_tensorflow_Session_run()
191 t[i] = reinterpret_cast<jlong>(output_values[i]); in Java_org_tensorflow_Session_run()
|