Home
last modified time | relevance | path

Searched refs:WriteToTensor (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dstring_util_test.cc73 buf0.WriteToTensor(t0, new_shape); in TEST()
180 buf.WriteToTensor(t0, nullptr); in TEST()
190 buf.WriteToTensor(t0, new_shape); in TEST()
H A Dstring_util.cc104 WriteToTensor(tensor, dims); in WriteToTensorAsVector()
107 void DynamicBuffer::WriteToTensor(TfLiteTensor* tensor, in WriteToTensor() function in tflite::DynamicBuffer
H A Dstring_util.h87 void WriteToTensor(TfLiteTensor* tensor, TfLiteIntArray* new_shape);
H A Dmodel_test.cc788 buf.WriteToTensor(tensor, /*new_shape=*/nullptr); in TEST()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dwhitespace_tokenizer.cc115 buffer.WriteToTensor(output_values, output_shape); in WritePaddedOutput()
148 buffer.WriteToTensor(output_values, output_shape); in WriteRaggedOutput()
H A Dngrams.cc190 buffer.WriteToTensor(output_values, /*new_shape=*/nullptr); in Eval()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtile.cc170 buffer->WriteToTensor(out_data, /*new_shape=*/nullptr); in TileStringOneDimension()
286 buffer.WriteToTensor(output, /*new_shape=*/nullptr); in Eval()
H A Dfill.cc121 buffer.WriteToTensor(output, /*new_shape=*/nullptr); in FillString()
H A Dgather.cc170 buffer.WriteToTensor(output, /*new_shape=*/nullptr); in GatherStrings()
H A Dsubgraph_test_util.cc655 buf.WriteToTensor(tensor, /*new_shape=*/TfLiteIntArrayCreate(0)); in FillScalarStringTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtensor.h30 ~SequentialTensorWriter() { buffer_.WriteToTensor(output_, nullptr); } in ~SequentialTensorWriter()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/
H A Dlookup_util.h105 void Commit() { buf_.WriteToTensor(values_, nullptr); } in Commit()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
H A Dsentencepiece_detokenizer_tflite.cc85 buf.WriteToTensor(&output_values, nullptr); in Eval()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/
H A Dtflite_inference_main.cc70 buffer.WriteToTensor(tensor, /*new_shape=*/nullptr); in RunWithRandomInputs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/
H A Dcalibration_wrapper.cc453 buffer.WriteToTensor(subgraph->tensor(index), /*new_shape=*/nullptr); in SetTensor()
533 buffer.WriteToTensor(interpreter_->tensor(index), /*new_shape=*/nullptr); in SetTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate.cc131 dynamic_buffer.WriteToTensor(output, /*new_shape=*/nullptr); in CopyFromBufferHandle()
H A Dtflite_subgraph_execute.cc179 dynamic_buffer.WriteToTensor(subgraph_input, /*new_shape=*/nullptr); in SetSubgraphInput()
H A Dtest_util.cc34 dynamic_buffer.WriteToTensor(interpreter_->tensor(tensor_index), in SetStringValues()
H A Dbuffer_map_test.cc75 b.WriteToTensor(tensor.get(), ConvertVectorToTfLiteIntArray(shape)); in MakeLiteTensor()
H A Dutil_test.cc254 buf.WriteToTensor(&tflite_tensor, nullptr); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtflite_tensor_view.cc118 buf.WriteToTensor(wrapped_tensor, /*new_shape=*/nullptr); in ~StringBuffer()
H A Dtest_util.cc185 buf.WriteToTensor(tensor, /*new_shape=*/nullptr); in PopulateTfLiteTensorValue()
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dtflite_wrapper.cc53 buf.WriteToTensor(tensor, nullptr); in AssignStringInput()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/benchmark/
H A Dbenchmark_tflite_model.cc641 ->WriteToTensor(t, /*new_shape=*/nullptr); in ResetInputsAndOutputs()
648 buffer.WriteToTensor(t, /*new_shape=*/nullptr); in ResetInputsAndOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/
H A Dtensor_jni.cc412 dst_buffer.WriteToTensor(tensor, /*new_shape=*/nullptr); in WriteMultiDimensionalStringArray()
464 dst_buffer.WriteToTensor(tensor, /*new_shape=*/nullptr); in WriteScalarString()

12