Home
last modified time | relevance | path

Searched refs:IsCpuCompatibleDataFormat (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/remapper/
H A Dremapping_helper.h157 bool IsCpuCompatibleDataFormat(Operation* conv_op,
206 IsCpuCompatibleDataFormat(contraction_op); in IsCpuCompatible()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc318 bool IsCpuCompatibleDataFormat(const RemapperContext& ctx, in IsCpuCompatibleDataFormat() function
352 IsCpuCompatibleDataFormat(ctx, conv2d); in IsCpuCompatibleConv2D()
358 IsCpuCompatibleDataFormat(ctx, conv3d); in IsCpuCompatibleConv3D()
850 !IsCpuCompatibleDataFormat(ctx, conv2d_node_def) || in FindConv2DWithBatchNorm()