Home
last modified time | relevance | path

Searched refs:set_input_tensors (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc681 c->set_input_tensors(input_tensors); in RunShapeFn()
779 c->set_input_tensors(input_tensors); in RunShapeFn()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils.cc346 c.set_input_tensors(input_tensors); in InferReturnTypeComponentsForTFOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc143 ic.set_input_tensors(input_tensors); in CompileToCustomCallCallingTfKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.h378 void set_input_tensors(const std::vector<const Tensor*>& input_tensors) { in set_input_tensors() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc647 c.set_input_tensors(input_tensors_vector); in ValidateOutputTensorShapeConsistency()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc1126 ic->set_input_tensors(input_tensors); in UpdateNode()