Home
last modified time | relevance | path

Searched defs:IsDeviceCompatible (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dfused_kernel_matcher.cc111 virtual bool IsDeviceCompatible(SrcOpT contraction_op, BiasAddOp bias_add, in IsDeviceCompatible() function in mlir::TF::__anon9059b8460111::FuseContractionWithBiasAdd
266 bool IsDeviceCompatible(Conv2DOp conv, BiasAddOp bias_add, in IsDeviceCompatible() function in mlir::TF::__anon9059b8460111::FuseConv2DBiasAdd
303 bool IsDeviceCompatible(MatMulOp matmul, BiasAddOp bias_add, in IsDeviceCompatible() function in mlir::TF::__anon9059b8460111::FuseMatMulBiasAdd
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/remapper/
H A Dremapping_helper.h217 bool IsDeviceCompatible(const Pattern& pattern) const { in IsDeviceCompatible() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc543 bool IsDeviceCompatible(const RemapperContext& ctx, Pattern& matched, in IsDeviceCompatible() function