Home
last modified time | relevance | path

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

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