- Sort Score
- Result 10 results
- Languages All
Results 181 - 190 of 380 for getLoc (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/guarantee_all_funcs_one_use.cc
"one use for all functions. Functions in the " "recursive cycle are: "; llvm::interleaveComma(*scci, err, [&](CallGraphNode *node) { err << node->getCallableRegion()->getLoc(); }); return err; } } return success(); } #define GEN_PASS_DEF_GUARANTEEALLFUNCSONEUSEPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
const auto composite_resource_users = GetResourceOpsUsingCompositeArgsInReplicate(replicate_op); for (auto resource_user : composite_resource_users) { WrapOpInLaunch(builder, resource_user->getLoc(), resource_user, tensorflow::GetDeviceAliasForLogicalCore(0)); } } void TPUColocateCompositeResourceOps::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
if (call_op_has_platform_index_arg) { arg_mapper.map(func_op.getArgument(0), builder.create<mhlo::ConstantOp>( func_op.getLoc(), builder.getI16IntegerAttr(0))); } for (auto [func_arg, operand] : llvm::zip_equal( func_op.getArguments().take_back(call_op.getNumOperands()), call_op.getOperands())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_jax_random.cc
if (result_tuple_ty.size() != 1) return; auto result_ty = mlir::dyn_cast<ShapedType>(result_tuple_ty.getType(0)); func.eraseBody(); func.addEntryBlock(); ImplicitLocOpBuilder builder(func.getLoc(), func.getBody()); llvm::SmallVector<int32_t> result_shape_i32; auto result_shape = result_ty.getShape(); for (auto element : result_shape) { result_shape_i32.push_back(static_cast<int32_t>(element)); }
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/lite/quantization/tensorflow/fallback_to_flex_ops.cc
return QuantizableOpsInDefaultMode().count(op_name) > 0; } else if (mode_ == kLegacyIntegerMode) { return QuantizableOpsInLegacyMode().count(op_name) > 0; } else { mlir::emitError(getOperation().getLoc(), "Unregconized mode: " + mode_); signalPassFailure(); return true; } } // Converts the operation to a TFLite Flex op. bool ConvertToFlexOp(Operation *op); };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
TF::PartitionedCallOp call_op) const { // If unit-wise quantization config is found, overwrite the default config. auto quantization_unit = FindQuantizationUnitFromLoc(call_op.getLoc()); if (!quantization_unit.has_value()) return std::nullopt; for (const auto& unit_config : quant_options_.unit_wise_quantization_specs()) { for (const auto& unit : unit_config.unit()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
std::replace(func_name.begin(), func_name.end(), '/', '_'); FunctionType func_type = FunctionType::get(context, input_types, result_types); Location loc = metadata.ops.front()->getLoc(); func::FuncOp func_op = func::FuncOp::create(loc, func_name, func_type); // Sets the device attribute for every input and every result of the // function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
if (batch_norm.getIsTraining()) return failure(); if (!batch_norm.getY().hasOneUse()) return failure(); // Build the newly fused operation to replace the batch norm OperationState state(batch_norm.getLoc(), _FusedBatchNormExOp::getOperationName()); state.addOperands(batch_norm.getOperands()); if (side_input) state.operands.push_back(side_input); state.addTypes(batch_norm.getResultTypes());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
def TrueBoolAttr : AttrConstraint<CPred<"$_self.cast<BoolAttr>().getValue()">>; def CastValueToI64: NativeCodeCall< "CastValueToI64($0.getLoc(), $1, &$_builder)">; def CastValueToElementType: NativeCodeCall< "$_builder.create<ConvertOp>($0.getLoc(), $1, " "getElementTypeOrSelf($2.getType()))">; // Here, $0 is an ElementsAttr with exactly one element of type integer. $1 is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
const DictionaryAttr main_arg_attr = src_func_op.getArgAttrDict(src_arg_idx); main_func_op.insertArgument(main_arg_idx, src_arg.getType(), main_arg_attr, src_arg.getLoc()); const std::string new_input_name = absl::StrCat(GetInitializerType(src_func_op), "_", src_arg_idx, ":0"); AddEntryFunctionInput(new_input_name, main_func_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0)