- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for lookupState (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/colocate_tpu_copy_with_dynamic_shape.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 23 00:30:27 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc
} if (llvm::dyn_cast_or_null<TF::VarHandleOp>(resource.getDefiningOp())) { return; // We're already directly after a VarHandleOp. } const TF::ResourceDataflowState* state = solver.lookupState<TF::ResourceDataflowState>(resource); if (!state) { // Can't actually happen. Even for gaps in the dataflow, we'll receive // the initial state. return; } auto ops = state->getValue().ops;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
resource = read.getResource(); } else if (auto write = llvm::dyn_cast<TF::AssignVariableOp>(op)) { resource = write.getResource(); } const TF::ResourceDataflowState* state = solver.lookupState<TF::ResourceDataflowState>(resource); if (!state) { return nullptr; } auto ops = state->getValue().ops; if (ops.size() != 1) { return nullptr; } Operation* source = *ops.begin();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
(void)id; } } for (auto& arg : op->getOpOperands()) { if (!IsResourceType(arg.get().getType())) { continue; } auto lattice = solver.lookupState<TF::IsCompositeDataflowState>(arg.get())->getValue(); bool is_read = read_array.contains(arg.getOperandNumber()); bool is_update = update_array.contains(arg.getOperandNumber());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0)