- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 102 for GetOperands (0.37 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
YieldOp yield = GetYield(); return wrapped_op.getNumResults() == yield.getNumOperands() && std::equal(wrapped_op.getResults().begin(), wrapped_op.getResults().end(), yield.getOperands().begin()); } mlir::LogicalResult IslandOp::verify() { IslandOp island = *this; if (!island.GetBody().args_empty()) return island.emitOpError() << "expects body without any arguments";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
Operation::operand_range mean(fused_batch_norm->getOperands()); ::mlir::FloatAttr exponential_avg_factor; ::mlir::TF::FusedBatchNormV3Op root; Operation::operand_range offset(fused_batch_norm->getOperands()); Operation::operand_range x(fused_batch_norm->getOperands()); Operation::operand_range scale(fused_batch_norm->getOperands()); Operation::operand_range variance(fused_batch_norm->getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
return std::nullopt; for (auto [arg, operand] : llvm::zip(block.getArguments(), call.getOperands())) { if (arg != operand) return std::nullopt; } for (auto [ret, operand] : llvm::zip(call.getResults(), yield.getOperands())) { if (ret != operand) return std::nullopt; } SymbolRefAttr symbol = call.getCallableForCallee().get<SymbolRefAttr>();
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/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
func_operand_types.push_back(operand.getType()); // Function results are the yield operands SmallVector<Type, 16> func_result_types; for (Value operand : island_op.GetYield().getOperands()) func_result_types.push_back(operand.getType()); FunctionType func_type = FunctionType::get(ctx, func_operand_types, func_result_types); // Create the outlined function
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
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/tpu_partitioned_op_conversion.cc
partition_dims[partition_dim] = num_cores_per_replica.getInt(); } if constexpr (is_input) { auto pi = builder.create<TF::TPUPartitionedInputV2Op>( op.getLoc(), op.getType(), op.getOperands(), builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false), op.get_XlaShardingAttr()); op->replaceAllUsesWith(pi); } else { auto po = builder.create<TF::TPUPartitionedOutputV2Op>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K 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/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/transforms/reduce_type_precision.cc
auto new_gather_op = rewriter.create<TFL::GatherOp>( op.getLoc(), /*result=*/ mlir::cast<TensorType>(op.getResult().getType()) .clone(builder.getI4Type()), /*operand=*/op.getOperands(), op->getAttrs()); rewriter.replaceAllUsesWith(op.getResult(), new_gather_op.getResult()); return success(); } }; class ReduceTypePrecisionPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0)