Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc545 bool FindConv2DWithBatchNorm(const RemapperContext& ctx, int node_index, in IsCpuCompatibleDataType() function
610 if (!FindConv2DWithBatchNorm(ctx, batch_norm_node_view->node_index(), &base)) in IsCpuCompatibleDataType()
1890 FindConv2DWithBatchNorm(ctx, i, &contract_with_batch_norm)) { in IsCpuCompatibleDataType()