- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for inner_loop (0.3 sec)
-
tensorflow/c/while_loop_test.cc
node_name = "test_loop/body/add"; EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr); node_name = "test_loop/body/inner_loop/body/one"; EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr); node_name = "test_loop/body/inner_loop/cond/less_than"; EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr); // Run the graph Run({0, 0}); ExpectOutputValue(0, 8);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 06:05:56 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
for (Operation& inner_op : llvm::make_early_inc_range(replicate_op.GetBody())) { if (llvm::isa<tf_device::ReturnOp>(inner_op)) continue; // Skip hoisting if the inner op device attribute is a virtual device // defined by tf_device.replicate. if (DescendantUsesVirtualDevice(virtual_device_list, &inner_op)) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/pin_ops_with_side_effects.cc
// We have to recurse, since we might have wrapped a side-effectful operator // in a tfl::CustomTfOp. if (op.walk([&](Operation *inner_op) { return op_has_side_effects_(inner_op) ? WalkResult::interrupt() : WalkResult::advance(); }).wasInterrupted()) { ops_with_side_effects.push_back(&op); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getSuccessors(), new_regions); inner_op = rewriter.create(state); } else { bool disable_call_shape_inference = false; if (op->hasAttr("_disable_call_shape_inference")) { disable_call_shape_inference = op->getAttrOfType<BoolAttr>("_disable_call_shape_inference") .getValue(); } inner_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
for (Operation* cluster_op : cluster_ops) { cluster_op->moveBefore(body, body->end()); cluster_op->walk([&](Operation* inner_op) { inner_op->removeAttr(mlir::TF::kReplicationInfoAttr); inner_op->removeAttr(mlir::TF::kCompileDeviceTypeAttr); if (auto attr = inner_op->getAttrOfType<StringAttr>(kDeviceAttr)) { // Preserve device attribute if the op is placed on a replicated core
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} else if (auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) { Operation& inner_op = island.GetBody().front(); auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef()); if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) { // If it is TF Control dialect specific op, look up custom operation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} else if (auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) { Operation& inner_op = island.GetBody().front(); auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef()); if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) { // If it is TF Control dialect specific op, look up custom operation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} auto attr_value = mlir::DenseI32ArrayAttr::get(inner_op->getContext(), values); inner_op->setAttr(attr_name, attr_value); }; if (inner_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>() || inner_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { // The op has multiple variadic operands or results.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
control_nodes_at.try_emplace(&item.value(), item.index()); } } for (auto outer_op : control_nodes) { auto control_node_op = dyn_cast<mlir::TFL::ControlNodeOp>(outer_op); auto* inner_op = &control_node_op.getBody().front().front(); auto control_token = control_node_op.getControl(); // Now go through all uses. Since *block is in executable order, control
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)