Home
last modified time | relevance | path

Searched defs:shape_inference (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/common/
H A Dtfl_pass_config.h69 bool shape_inference; member
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference_testutil.cc25 namespace shape_inference { namespace
H A Dshape_inference_testutil_test.cc25 namespace shape_inference { namespace
H A Dcommon_shape_fns_test.cc28 namespace shape_inference { namespace
H A Dshape_inference.cc28 namespace shape_inference { namespace
H A Dshape_inference_test.cc30 namespace shape_inference { namespace
H A Dcommon_shape_fns.cc30 namespace shape_inference { namespace
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dstring_ops.cc30 namespace shape_inference { namespace
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dlazy_ir.py494 def shape_inference(self, func: NativeFunction, schema: LazyIrSchema) -> str: member in GenLazyNativeFuncDefinition