Home
last modified time | relevance | path

Searched refs:input_tensors_vector (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc622 std::vector<const tensorflow::Tensor*> input_tensors_vector(num_inputs, in ValidateOutputTensorShapeConsistency() local
627 input_tensors_vector, {}, in ValidateOutputTensorShapeConsistency()
637 input_tensors_vector[i] = in ValidateOutputTensorShapeConsistency()
647 c.set_input_tensors(input_tensors_vector); in ValidateOutputTensorShapeConsistency()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc635 std::vector<const Tensor*> input_tensors_vector(num_inputs, nullptr); in TFE_InferShapes() local
650 input_tensors_vector[i] = &input_tensor; in TFE_InferShapes()
656 std::vector<ShapeHandle>(num_inputs), input_tensors_vector, in TFE_InferShapes()