- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 129 for getFlow (0.19 sec)
-
tensorflow/compiler/mlir/lite/metrics/error_collector_inst.cc
auto collectOps = [this](Operation *op) { const auto &op_name = op->getName().getStringRef().str(); if (absl::StartsWith(op_name, "tf.") || absl::StartsWith(op_name, "tfl.")) { loc_to_name_.emplace(op->getLoc(), op_name); } }; for (auto ®ion : module->getRegions()) { region.walk(collectOps); } pass_name_ = extract_pass_name(pass->getName().str()); error_collector_->Clear(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/cast_bf16_ops_to_f32.cc
for (int i = 0; i < op->getNumOperands(); i++) { Value input = op->getOperand(i); if (getElementTypeOrSelf(input).isBF16()) { Value f32_cast = rewriter.create<TF::CastOp>( op->getLoc(), CloneTypeWithNewElementType(input.getType(), rewriter.getF32Type()), input); op->setOperand(i, f32_cast); } } // Casts BF16 outputs of the operation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
auto new_reshape_type = RankedTensorType::get(new_reshape_dims, el_ty); ReshapeOp new_reshape = rewriter.create<ReshapeOp>(new_reshape_shape.getLoc(), new_reshape_type, op.getInput(), new_reshape_shape); TF::ConstOp new_broadcast_shape = GetI64ConstantTensor(rewriter, reshape_shape, op.getLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
for (Value result : new_launch_op_results) new_launch_op_results_types.push_back(result.getType()); auto new_launch_op = builder.create<tf_device::LaunchOp>( old_launch_op->getLoc(), old_launch_op->getDeviceAttr(), /*result_types=*/new_launch_op_results_types); new_launch_op.getBody().takeBody(old_launch_op->getBody());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc
// When creating the new const op, use a fused location that combines the // original const and the qbarrier that led to the quantization. auto fusedLoc = rewriter.getFusedLoc( {qbarrier.getArg().getDefiningOp()->getLoc(), qbarrier.getLoc()}); auto newConstOp = rewriter.create<arith::ConstantOp>( fusedLoc, newConstValueType, cast<TypedAttr>(newConstValue)); rewriter.replaceOpWithNewOp<StorageCastOp>(qbarrier, qbarrier.getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
src/sync/pool.go
// reuse. x, _ = l.shared.popHead() if x == nil { x = p.getSlow(pid) } } runtime_procUnpin() if race.Enabled { race.Enable() if x != nil { race.Acquire(poolRaceAddr(x)) } } if x == nil && p.New != nil { x = p.New() } return x } func (p *Pool) getSlow(pid int) any { // See the comment in pin regarding ordering of the loads.
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Jun 07 21:14:51 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
other.addEntryBlock(); OpBuilder builder(other.getRegion()); auto call_op = builder.create<mlir::func::CallOp>( f.getLoc(), f.getFunctionType().getResults(), f.getSymName(), other.getRegion().getArguments()); builder.create<mlir::func::ReturnOp>(f.getLoc(), call_op.getResults()); } Unexport(f); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
// be created and updated, rather than updating the old op. Thus we use an // OperationState so we can add regions to the new op. OperationState state(op->getLoc(), op->getName().getStringRef(), operands, new_results, op->getAttrs(), op->getSuccessors()); for (Region& region : op->getRegions()) { Region& new_region = *state.addRegion();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
returned_type = quant::ConvertSignedQuantizedToUnsigned( dequantize_input.getType(), dequantize_op.getLoc()); // replace the dequantize op by a quantize op TypeAttr type_attr = TypeAttr::get(returned_type); auto quantize_op = builder.create<QuantizeOp>( dequantize_op.getLoc(), returned_type, dequantize_input, type_attr); returned_value = quantize_op.getOutput(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
builder.create<tf_saved_model::SessionInitializerOp>( module->getLoc(), builder.getArrayAttr(func_names)); // Create AssetOp; this holds the checkpoint_path. // TODO(b/318761632): Cleanup usage of string literals, instead use constants. auto asset_op = builder.create<tf_saved_model::AssetOp>( module->getLoc(), /*sym_name=*/
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0)