Home
last modified time | relevance | path

Searched defs:ShapeInference (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference_pass.cc33 class ShapeInference : public TensorFlowShapeInferencePassBase<ShapeInference> { class
H A Dshape_inference.cc566 class ShapeInference { class
824 ShapeInference::ShapeInference(int64_t graph_version, ModuleOp module, in ShapeInference() function in mlir::TF::ShapeInference
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtf_op_shim.h111 static tensorflow::Status ShapeInference( in ShapeInference() function
H A Dop_kernel.h206 static absl::Status ShapeInference(ShapeInferenceContext* ctx) { in ShapeInference() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
H A Dsimple_op.h134 static absl::Status ShapeInference(ShapeInferenceContext* ctx) { in ShapeInference() function
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/
H A Dpass.cc63 class ShapeInference : public ShapeInferenceBase<ShapeInference> { class