| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 168 T* __restrict output_values) { in ConcatV2Args2() argument
175 std::copy(arg0_values, arg0_values + arg0_line_size, output_values); in ConcatV2Args2()
177 output_values += arg0_line_size; in ConcatV2Args2()
178 std::copy(arg1_values, arg1_values + arg1_line_size, output_values); in ConcatV2Args2()
180 output_values += arg1_line_size; in ConcatV2Args2()
189 T* __restrict output_values) { in MatMul() argument
196 auto result = MatrixMap<T>(output_values, weight_shape[1], input_shape[0]); in MatMul()
210 *output_values++ = value; in MatMul()
225 T* __restrict output_values) { in DepthwiseConv2dNative() argument
257 T* out_write_ptr = output_values; in DepthwiseConv2dNative()
[all …]
|
| D | onedevice_train_palm_detection_filter_inference.cc | 169 T* __restrict output_values) { in ConcatV2Args2() argument
176 std::copy(arg0_values, arg0_values + arg0_line_size, output_values); in ConcatV2Args2()
178 output_values += arg0_line_size; in ConcatV2Args2()
179 std::copy(arg1_values, arg1_values + arg1_line_size, output_values); in ConcatV2Args2()
181 output_values += arg1_line_size; in ConcatV2Args2()
190 T* __restrict output_values) { in MatMul() argument
197 auto result = MatrixMap<T>(output_values, weight_shape[1], input_shape[0]); in MatMul()
211 *output_values++ = value; in MatMul()
226 T* __restrict output_values) { in DepthwiseConv2dNative() argument
258 T* out_write_ptr = output_values; in DepthwiseConv2dNative()
[all …]
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 168 T* __restrict output_values) { in ConcatV2Args2() argument
175 std::copy(arg0_values, arg0_values + arg0_line_size, output_values); in ConcatV2Args2()
177 output_values += arg0_line_size; in ConcatV2Args2()
178 std::copy(arg1_values, arg1_values + arg1_line_size, output_values); in ConcatV2Args2()
180 output_values += arg1_line_size; in ConcatV2Args2()
189 T* __restrict output_values) { in MatMul() argument
196 auto result = MatrixMap<T>(output_values, weight_shape[1], input_shape[0]); in MatMul()
210 *output_values++ = value; in MatMul()
225 T* __restrict output_values) { in DepthwiseConv2dNative() argument
257 T* out_write_ptr = output_values; in DepthwiseConv2dNative()
[all …]
|
| /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()
178 int32* output_indexes, T* output_values) { in TopK() argument
189 T* output_row = output_values + row * k; in TopK()
207 TfLiteTensor* output_values; in Prepare() local
209 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in Prepare()
210 TF_LITE_ENSURE_TYPES_EQ(context, input->type, output_values->type); in Prepare()
224 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 &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_slice_op_gpu.cu.cc | 94 int64_t* __restrict__ output_indices, T* __restrict__ output_values) { in SparseSliceGatherKernel() argument
99 output_values[output_nz] = input_values[input_nz]; in SparseSliceGatherKernel()
156 Tensor* output_values = nullptr; in operator ()() local
158 context, context->allocate_output(1, {0}, &output_values), done); in operator ()()
223 Tensor* output_values = nullptr; in operator ()() local
225 context, context->allocate_output(1, {output_nnz}, &output_values), in operator ()()
227 T* output_values_ptr = output_values->vec<T>().data(); in operator ()()
|
| D | sparse_fill_empty_rows_op_gpu.cu.cc | 132 const Tindex* num_new_rows_before, Tindex* output_indices, T* output_values, in ScatterInputElementsKernel() argument
141 output_values[output_i] = values[input_i]; in ScatterInputElementsKernel()
154 const bool* empty_row_indicator, Tindex* output_indices, T* output_values) { in ScatterNewElementsKernel() argument
162 output_values[output_i] = *default_value; in ScatterNewElementsKernel()
191 T* output_values; in operator ()() local
195 &output_values, &reverse_index_map)); in operator ()()
375 T* output_values; in operator ()() local
380 context, N, rank, num_empty_rows, &output_indices, &output_values, in operator ()()
404 output_values, reverse_index_map), in operator ()()
414 output_values), in operator ()()
[all …]
|
| D | sparse_split_op_gpu.cu.cc | 170 T* __restrict__* __restrict__ output_values = in SparseSplitScatterKernel() local
180 output_values[slice_index][slice_nz] = input_values[input_nz]; in SparseSplitScatterKernel()
314 GpuDeviceArrayOnHost<T*> output_values(context, num_split); in operator ()() local
319 &output_indices, &output_values), in operator ()()
330 input_values_ptr, output_indices.data(), output_values.data()), in operator ()()
348 GpuDeviceArrayOnHost<T*>* output_values) const { in AllocateOutputs()
350 TF_RETURN_IF_ERROR(output_values->Init()); in AllocateOutputs()
362 output_values->Set(slice_index, output_vals->vec<T>().data()); in AllocateOutputs()
373 TF_RETURN_IF_ERROR(output_values->Finalize()); in AllocateOutputs()
|
| D | sparse_tensors_map_ops.cc | 308 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
311 auto output_values_t = output_values.vec<T>(); in Compute()
322 SparseTensor::Create(output_indices, output_values, in Compute()
400 const Tensor& output_values = st.values(); in Compute() local
408 OP_REQUIRES(context, TensorShapeUtils::IsVector(output_values.shape()), in Compute()
412 output_values.shape().DebugString())); in Compute()
414 context, DataTypeToEnum<T>::value == output_values.dtype(), in Compute()
418 "].values.dtype() == ", DataTypeString(output_values.dtype()))); in Compute()
421 OP_REQUIRES(context, num_entries == output_values.dim_size(0), in Compute()
426 " vs. ", output_values.dim_size(0))); in Compute()
[all …]
|
| /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_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()
|
| 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()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | batchmatmul.py | 89 output_values = sess.run(
91 return [input0_value], output_values
97 output_values = sess.run(
99 return [input0_value, input1_value], output_values
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | batchmatmul.py | 91 output_values = sess.run(
93 return [input0_value], output_values
99 output_values = sess.run(
101 return [input0_value, input1_value], output_values
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 163 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
164 if (!output_values || !output_values.getElementType().isa<StringType>()) { in VerifyNgrams()
169 if (input_values.hasRank() && output_values.hasRank() && in VerifyNgrams()
170 input_values.getRank() != output_values.getRank()) { in VerifyNgrams()
181 auto output_values = GetResultType(func, kValues); in VerifyNgrams() local
182 if (!RankEquals(output_values, 1) || in VerifyNgrams()
183 !output_values.getElementType().isa<StringType>()) { in VerifyNgrams()
|
| /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"
|