Searched refs:FindContractionWithBias (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 401 bool FindContractionWithBias(const RemapperContext& ctx, int node_index, in IsCpuCompatibleDataType() function
459 if (!FindContractionWithBias(ctx, bias_add_node_view->node_index(), &base, in IsCpuCompatibleDataType()
644 if (!FindContractionWithBias(ctx, bias_add_node_view->node_index(), base, in IsCpuCompatibleDataType()
1852 FindContractionWithBias(ctx, i, &contract_with_bias)) { in IsCpuCompatibleDataType()
|