Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc188 struct ContractionWithSqueezeAndBiasAdd { struct
189 ContractionWithSqueezeAndBiasAdd() = default;
190 ContractionWithSqueezeAndBiasAdd(int contraction, int squeeze, int bias_add) in ContractionWithSqueezeAndBiasAdd() function
536 const ContractionWithSqueezeAndBiasAdd& matched, in IsGpuCompatible()
754 ContractionWithSqueezeAndBiasAdd* matched) { in FindConvWithSqueezeAndBias()
800 const ContractionWithSqueezeAndBiasAdd pattern{ in FindConvWithSqueezeAndBias()
2307 const ContractionWithSqueezeAndBiasAdd& matched, in AddFusedConvNode()
3624 ContractionWithSqueezeAndBiasAdd contract_with_squeeze_and_bias; in Optimize()