- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 90 for eraseOp (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
attr->contains("Truncate") && !attr->at("Truncate").b()) { node_def->set_op("Identity"); attr->insert({{"T", attr->at("SrcT")}}); attr->erase("SrcT"); attr->erase("DstT"); attr->erase("Truncate"); } } } // namespace Status GetAttrValuesFromOperation( mlir::Operation* inst, llvm::StringRef name, const tensorflow::OpRegistrationData* op_reg_data,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
/*operands=*/merged_else_yield_values); } // Merge the IfRegions in each segment. In the meantime, the old IfRegions in // the segment will be added to `regions_to_remove`. They will be erased in the // end. // `if_op_segment` refers to segments of IfRegions. `moved_op_list` refers to // the ops to be moved after new merged IfRegion. `regions_to_remove` refers to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
// we replace that block-argument's uses with the same (implicitly captured) // token 'token', used for If op, and erase the argument. // Note that 'RewriteControlFlowOpRegion' sets the token, used for the first // operation of region_idx'th region, to the dummy block-argument. As we // erase that argument, we also need to make sure that the token used for // the next operation is set to 'token'.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
op->getLoc(), DenseIntElementsAttr::get( RankedTensorType::get({}, builder.getI64Type()), {device_ordinal})); op->replaceAllUsesWith(const_op); op->erase(); return WalkResult::advance(); } if (!devices.has_value()) return WalkResult::advance(); // Map aliased devices to explicit devices based on replica.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
} } state.addOperands(operands); Operation* new_op = builder.create(state); item.replaceAllUsesWith(new_op); new_op->setAttrs(item.getAttrDictionary()); item.erase(); } AddStatefulControlFlowDependencies(graph_op); } // Populates an empty IslandOp and with a NoOp or Identity/IdentityN depending // on if there are any data results.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen_test.cc
// Get rid of all CR characters, we may be running under windows. string sanitized_expected_contents(expected_contents); if (ignore_cr) { sanitized_expected_contents.erase( std::remove(sanitized_expected_contents.begin(), sanitized_expected_contents.end(), '\r'), sanitized_expected_contents.end()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 01 02:13:40 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
then_name, else_name, if_region.getIsStateless()); CopyAndOverrideAttributes(if_region, if_op, &builder); if_region.replaceAllUsesWith(if_op.getResults()); if_region.erase(); if (to_bool && to_bool.use_empty()) to_bool.erase(); return success(); } // Transform CaseRegion to Case. LogicalResult RegionControlFlowToFunctional::ConvertCaseOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
auto s_const = builder.create<SparseConstOp>( op->getLoc(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_const.getResult()); cst.erase(); } else if (type.getElementType().isF16()) { std::vector<Eigen::half> dense_data; dense_data.reserve(type.getNumElements()); for (const auto& val : attr.getValues<APFloat>())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
it->second--; if (it->second != 0) { return; } tensor_usage_.erase(it); auto tensor_op_it = tensor_tape_.find(tensor_id); if (tensor_op_it == tensor_tape_.end()) { return; } const int64_t op_id = tensor_op_it->second; if (op_id == -1) { // Do not delete watched tensors. return; } tensor_tape_.erase(tensor_op_it); auto op_it = op_tape_.find(op_id);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} args.append(new_args.begin(), new_args.end()); } else { args.append(yield_op->operand_begin(), yield_op->operand_end()); } b.create<func::ReturnOp>(yield_op->getLoc(), args); yield_op->erase(); SymbolTable(region.getParentOfType<ModuleOp>()).insert(outlined_func); outlined_func.setPrivate(); return outlined_func; } // Replace region with call to outline function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0)