Home
last modified time | relevance | path

Searched refs:ResizeInputTensorStrict (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dbert_nl_classifier.cc92 GetTfLiteEngine()->interpreter()->ResizeInputTensorStrict(kIdsTensorIndex, in Preprocess()
94 GetTfLiteEngine()->interpreter()->ResizeInputTensorStrict(kMaskTensorIndex, in Preprocess()
96 GetTfLiteEngine()->interpreter()->ResizeInputTensorStrict(kSegmentIdsTensorIndex, in Preprocess()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dsignature_runner.cc63 TfLiteStatus SignatureRunner::ResizeInputTensorStrict( in ResizeInputTensorStrict() function in tflite::SignatureRunner
70 return subgraph_->ResizeInputTensorStrict(it->second, new_size); in ResizeInputTensorStrict()
H A Dinterpreter.cc216 TfLiteStatus Interpreter::ResizeInputTensorStrict( in ResizeInputTensorStrict() function in tflite::Interpreter
218 return primary_subgraph().ResizeInputTensorStrict(tensor_index, dims); in ResizeInputTensorStrict()
H A Dsignature_runner.h128 TfLiteStatus ResizeInputTensorStrict(const char* input_name,
H A Dinterpreter_test.cc567 ASSERT_EQ(interpreter.ResizeInputTensorStrict(t, {1, 1, 3}), kTfLiteOk); in TEST()
572 ASSERT_EQ(interpreter.ResizeInputTensorStrict(t, {1, 2, 3}), kTfLiteError); in TEST()
598 ASSERT_EQ(interpreter.ResizeInputTensorStrict(t, {1, 2, 3}), kTfLiteOk); in TEST()
602 ASSERT_EQ(interpreter.ResizeInputTensorStrict(t, {1, 2, 4}), kTfLiteError); in TEST()
H A Dinterpreter.h454 TfLiteStatus ResizeInputTensorStrict(int tensor_index,
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/
H A Dcalibration_wrapper.cc441 TFLITE_PY_CHECK(subgraph->ResizeInputTensorStrict(index, dims)); in SetTensor()
521 TFLITE_PY_CHECK(interpreter_->ResizeInputTensorStrict(index, dims)); in SetTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_api_experimental.cc158 return signature_runner->impl->ResizeInputTensorStrict(input_name, dims); in TfLiteSignatureRunnerResizeInputTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsparse_to_dense_test.cc65 interpreter_->ResizeInputTensorStrict(output_shape_, output_shape_shape); in SparseToDenseOpModel()
H A Dconcatenation_test.cc522 interpreter_->ResizeInputTensorStrict(concat_inputs_[i], in PersistentConcatenationOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.h233 TfLiteStatus ResizeInputTensorStrict(int tensor_index,
H A Dsubgraph.cc924 TfLiteStatus Subgraph::ResizeInputTensorStrict(int tensor_index, in ResizeInputTensorStrict() function in tflite::Subgraph
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/
H A Dnativeinterpreterwrapper_jni.cc573 status = interpreter->ResizeInputTensorStrict( in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
H A Dinterpreter_wrapper.cc359 ->ResizeInputTensorStrict(i, dims)); in ResizeInputTensor()