Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc112 struct ContractionWithBiasAdd { struct
113 ContractionWithBiasAdd() = default;
114 ContractionWithBiasAdd(int contraction, int bias_add) in ContractionWithBiasAdd() argument
357 const ContractionWithBiasAdd& matched) { in IsCpuCompatibleDataType()
402 ContractionWithBiasAdd* matched, in IsCpuCompatibleDataType()
430 const ContractionWithBiasAdd pattern{contraction_node_view->node_index(), in IsCpuCompatibleDataType()
458 ContractionWithBiasAdd base; in IsCpuCompatibleDataType()
636 ContractionWithBiasAdd* base) { in IsCpuCompatibleDataType()
689 ContractionWithBiasAdd base; in IsCpuCompatibleDataType()
1060 const ContractionWithBiasAdd& matched, in IsCpuCompatibleDataType()
[all …]