- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for RemoveIdentityCast (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
// the `Cast` will be preserved. This transform is needed for the graph to be // executed on TPU or GPU devices, which do not have `Cast` registered as a // runtime OpKernel. void RemoveIdentityCast(NodeDef* node_def) { auto attr = node_def->mutable_attr(); if (node_def->op() == "Cast" && attr->contains("SrcT") && attr->contains("DstT") && attr->at("SrcT").type() == attr->at("DstT").type() &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)