Searched refs:RemapperContext (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 78 struct RemapperContext { struct
79 explicit RemapperContext(GrapplerItem* item, Status* status) in RemapperContext() argument
210 bool IsInPreserveSet(const RemapperContext& ctx, const NodeDef* node) { in IsInPreserveSet()
304 bool IsCpuCompatible(const RemapperContext& ctx, const Pattern& matched) { in IsCpuCompatibleDataType()
325 bool IsGpuCompatible(const RemapperContext& ctx, in IsCpuCompatibleDataType()
356 bool IsGpuCompatible(const RemapperContext& ctx, in IsCpuCompatibleDataType()
360 bool IsGpuCompatible(const RemapperContext& ctx, in IsCpuCompatibleDataType()
367 bool IsDeviceCompatible(const RemapperContext& ctx, Pattern& matched) { in IsCpuCompatibleDataType()
401 bool FindContractionWithBias(const RemapperContext& ctx, int node_index, in IsCpuCompatibleDataType()
442 const RemapperContext& ctx, int node_index, in IsCpuCompatibleDataType()
[all …]
|