- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 142 for GetOperands (0.14 sec)
-
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
// TODO(fengliuai): make build_list op only take tensor argument llvm::SmallVector<Attribute, 4> list_input_types; llvm::SmallVector<Value, 4> list_inputs; for (auto list_input : list_op.getOperands()) { auto cast_op = dyn_cast_or_null<CastOp>(list_input.getDefiningOp()); if (!cast_op) return failure(); list_inputs.push_back(cast_op.getArg());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
bool AddAccessedResourceIds( Operation* op, const mlir::TF::ResourceAliasAnalysis::Info& resource_analysis_info, llvm::SmallDenseSet<int64_t>& resource_ids) { for (Value operand : TF::filter_resources(op->getOperands())) { if (resource_analysis_info.IsUnknownResource(operand)) { VLOG(2) << " unknown access"; return true; } const auto& ids = resource_analysis_info.GetResourceUniqueIds(operand);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
tfl::SparseQConstOp, mlir::TFL::NoValueOp, mlir::stablehlo::ConstantOp, mlir::vhlo::ConstantOpV1>(op); } static bool IsTFResourceOp(Operation* op) { for (const auto& operand : op->getOperands()) { auto elementType = getElementTypeOrSelf(operand.getType()); if (mlir::isa<mlir::TF::ResourceType>(elementType)) { return true; } } for (const auto& result : op->getResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
IslandOp const_wrapper = GetDummyConstant(builder, const_type, loc); // Get new operand and result types. auto new_operands = llvm::to_vector<4>(while_op->getOperands()); auto new_result_types = llvm::to_vector<4>(while_op->getResultTypes()); Value const_output = const_wrapper.getOutputs()[0]; for (int i = 0; i < num_resource_inputs; ++i) { new_operands.push_back(const_output);
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/tpu_space_to_depth_pass.cc
auto maybe_replicate = llvm::dyn_cast<tf_device::ReplicateOp>(cluster_func->getParentOp()); llvm::SmallVector<int64_t, 8> transform_input_indices; for (const auto& input : llvm::enumerate(cluster_func.getOperands())) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(input.value())) { if (block_arg.getArgNumber() != arg_num) continue; // For a block argument, consider transforms only when it is a replicated
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
for (int i = 0; i < num_cores_per_replica; ++i) input_list->emplace_back(llvm::SmallVector<mlir::Value, 4>()); llvm::SmallVector<mlir::Value, 4> cluster_func_inputs( cluster_func.getOperands()); auto sharding_attrs = cluster_func.getOperation()->getAttrOfType<mlir::ArrayAttr>( kInputShardingAttr); // If sharding attribute does not exist, then all inputs are placed on 0th
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
mlir::Operation& return_op = region.back().back(); mlir::Location loc = return_op.getLoc(); op_builder.setInsertionPointToEnd(®ion.back()); op_builder.create<mlir::stablehlo::ReturnOp>(loc, return_op.getOperands()); return_op.erase(); } void InlineVhloOpRegion(mlir::Region& region, mlir::func::FuncOp func) { OpBuilder op_builder{region}; mlir::IRMapping mapper;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
} mhlo_op = rewriter.create<DstOpT>(loc, TypeRange(while_result_types), adaptor.getOperands()); } else { mhlo_op = rewriter.create<DstOpT>(loc, op.getResultTypes(), adaptor.getOperands()); } int64_t num_regions = op.getNumRegions(); for (int64_t idx = 0; idx < num_regions; ++idx) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
// `tf.While` or `tf.If` ops which are not part of the op definition but // appear in a variadic input list. add_remaining_effects(filter_resources(op->getOperands())); add_remaining_effects(filter_resources(op->getResults())); if (!found_any_effect) { // We haven't collected any side effect but the op is potentially
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
Type output_type = op.getResult().getType(); uint32_t axis = CastI64ToI32(op.getDimension()).value(); rewriter.replaceOpWithNewOp<TFL::ConcatenationOp>( op, output_type, op.getOperands(), axis, /*fused_activation_function=*/rewriter.getStringAttr("NONE")); } }; // Rewrites quantized stablehlo.pad to tfl.padv2. // tfl.dilate is introduced in between when interior padding exists.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)