- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 218 for getOperands (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
if (!graph) return false; Operation* terminator = block.getTerminator(); if (graph.getNumResults() != terminator->getNumOperands()) return false; for (auto result : llvm::zip(graph.getResults(), terminator->getOperands())) if (std::get<0>(result) != std::get<1>(result)) return false; return true; } // Checks if an operation of the tf_executor dialect can have TPU devices // propagated through.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
OpBuilder &builder) { llvm::SmallVector<Value> non_resource_args, resource_args; bool has_resources = false, in_order = true; for (const Value &arg : cluster_func_op.getOperands()) { if (!mlir::isa<TF::ResourceType>(getElementTypeOrSelf(arg.getType()))) { non_resource_args.push_back(arg); if (has_resources) in_order = false; } else { resource_args.push_back(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
module.walk([&](tf_device::LaunchFuncOp launch) { OpBuilder builder(launch); auto call_op = builder.create<TF::PartitionedCallOp>( module.getLoc(), launch.getResultTypes(), launch.getOperands(), SymbolRefAttr::get(builder.getContext(), launch.getFunc()), /*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr(""));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
Operation* terminator = execute_block.getTerminator(); builder->setInsertionPoint(terminator); auto yield = builder->create<tf_executor::YieldOp>( terminator->getLoc(), terminator->getOperands()); terminator->erase(); // Create new island for each region. builder->setInsertionPoint(island_op); auto execute_island = builder->create<tf_executor::IslandOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
has_resource_result = true; int result_idx = result.getResultNumber(); Value ret0 = op->getRegion(0).front().getTerminator()->getOperand(result_idx); for (Region ®ion : op->getRegions().drop_front()) { Value ret = region.front().getTerminator()->getOperand(result_idx); if (ret != ret0) { return op->emitError("Result #") << result_idx
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/tf_dataflow.h
} else if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) { for (auto ®ion : while_op->getRegions()) { for (auto [arg, value] : llvm::zip(region.getArguments(), while_op->getOperands())) { this->join(getLatticeElement(arg), *getLatticeElement(value)); } } } else if (auto while_op = dyn_cast<TF::WhileOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.cc
// Recreate the operation by using the wrapper's operands and return types. // TODO(fengliuai): copy the regions. OperationState state(op->getLoc(), op->getName().getStringRef(), parent_op->getOperands(), parent_op->getResultTypes(), op->getAttrs(), op->getSuccessors()); Operation* inlined = builder.create(state); parent_op->replaceAllUsesWith(inlined); parent_op->erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
for (auto operand_and_idx : llvm::enumerate(op->getOperands())) TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(operand_and_idx.value(), dst_node, operand_and_idx.index())); operand_offset = op->getNumOperands(); } // For all other ops (including tf_executor.island), add remaining edges. for (auto operand_and_idx : llvm::enumerate(inst->getOperands())) TF_RETURN_IF_ERROR(
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
for (auto operand_and_idx : llvm::enumerate(op->getOperands())) TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(operand_and_idx.value(), dst_node, operand_and_idx.index())); operand_offset = op->getNumOperands(); } // For all other ops (including tf_executor.island), add remaining edges. for (auto operand_and_idx : llvm::enumerate(inst->getOperands())) TF_RETURN_IF_ERROR(
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/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
SmallVector<NamedAttribute> attrs{op->getAttrs()}; attrs.push_back(call_target); auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), attrs); rewriter.replaceOp(op, custom_call.getResults()); return success(); } } // namespace template <typename OpTy> class SmuggleOpPattern : public OpRewritePattern<OpTy> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0)