Searched defs:node_device (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops.cc | 114 auto node_device = [&](Operation *n) -> StringAttr { in MultiDeviceProcessInlinedCallBlocks() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 1218 absl::string_view node_device, absl::string_view input_name, in AddNodeReshape()
1242 absl::string_view node_device, bool node_in_frame, int num_channels, in AddNodeShapeConst()
1284 const string& node_device = node->GetDevice(); in MaybeReshapeVectorFanin() local
|
| H A D | dependency_optimizer.cc | 655 string rest, node_device = node->device(); in GroupCrossDeviceControlEdges() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils.cc | 480 absl::string_view node_op, absl::string_view node_device, in IsKernelRegisteredForNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/ |
| H A D | debug_data.py | 1292 def node_device(self, node_name): member in DebugDumpDir
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 1475 StringPiece node_op, StringPiece node_device, AttrSlice node_attrs, in FindKernelDef()
|