- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 270 for getFlow (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
cluster_func->getLoc(), partitioned_output_types, result, partitioned_input.getPartitionDimsAttr(), partitioned_input.get_XlaShardingAttr()); for (auto [i, value] : llvm::enumerate(partitioned_output.getOutput())) { const auto& resource = packed_input ? inputs[0] : inputs[i]; builder.create<TF::AssignVariableOp>( assign_var->getLoc(), /*resource=*/resource, /*value=*/value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
maxs.push_back(rmax); } quant_type = quantfork::fakeQuantAttrsToType( op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range, expressed, is_signed); if (legacy_float_scale) { quant_type = DownCastScale(quant_type, mins, maxs, op->getLoc()); } } else if (auto stats = op.getLayerStats().dyn_cast<DenseFPElementsAttr>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
ShapedType shaped_type = RankedTensorType::get({1}, TF::StringType::get(builder.getContext())); auto const_op = builder.create<TF::ConstOp>( asset.getLoc(), DenseStringElementsAttr::get(shaped_type, {filename})); for (auto init_op : init_table_from_text_file_ops_to_erase) { // Replace the InitializeTableFromTextFileV2Op to use the saved model's
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc
mlir::quant::QuantizedType::castToExpressedType(input_type); builder->setInsertionPoint(op); auto dequantize_op = builder->create<TFL::DequantizeOp>( op->getLoc(), dequantized_input_type, input.get()); dequantized_inputs.push_back(dequantize_op); } else { dequantized_inputs.push_back(input.get()); } } // Result types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_weight.cc
rewriter.setInsertionPointAfter(op); // create new F16 constant op in that location ConstantOp new_const = rewriter.create<ConstantOp>( op->getLoc(), new_result_type, new_value_attr); ConvertOp dcast = rewriter.create<ConvertOp>(op->getLoc(), old_result_type, new_const); // replace all convert ops with dq op. convert_op->replaceAllUsesWith(dcast);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
auto dynamic_key = rewriter.create<TF::_XlaCompileMlirPlaceholderProgramKeyOp>( func.getLoc(), /*program=*/result_type, llvm::ArrayRef<Value>{}); auto recv_at_host = rewriter.create<TF::_XlaRecvAtHostOp>( func.getLoc(), op.getOperandTypes(), /*dynamic_key=*/dynamic_key, op.getSendKeyAttr(), /*device_ordinal=*/rewriter.getI64IntegerAttr(0),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
mlir::isa<quant::QuantizedType>(element_type)) { return; } op_builder->setInsertionPoint(const_op); Value scalar = op_builder->create<mhlo::ConstantOp>( const_op->getLoc(), DenseElementsAttr::get( RankedTensorType::get(/*shape=*/{}, element_type), splat_elements_attr.getSplatValue<Attribute>())); auto broadcast_dims = DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
BlockArgument param = callee.getArgument(i); if (mlir::isa<TF::ResourceType>(getElementTypeOrSelf(param.getType()))) { removed_params.set(i); callee.getBody().addArgument(param.getType(), param.getLoc()); param.replaceAllUsesWith(callee.getArguments().back()); removed_params.push_back(false); } } // Remove old resource-type parameters. callee.getBody().front().eraseArguments(removed_params);
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/quantization/tensorflow/passes/quantize.cc
auto fcast_op = rewriter.create<TF::CastOp>( preceding_sc_op->getLoc(), dq_arg_type.clone(rewriter.getF32Type()), preceding_sc_op.getResult()); // Create a new AvgPool op with float type. TF::AvgPoolOp float_avg_pool_op = rewriter.create<TF::AvgPoolOp>( avg_pool_op->getLoc(), avg_pool_op.getType().clone(rewriter.getF32Type()), /*operands=*/fcast_op.getResult(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/op_or_arg_name_mapper.cc
auto name_from_loc = mlir::GetNameFromLoc(op->getLoc()); if (!name_from_loc.empty()) return name_from_loc; // If the location is none of the expected types, then simply use name // generated using the op type. return std::string(op->getName().getStringRef()); } auto val = op_or_val.dyn_cast<mlir::Value>(); auto name_from_loc = mlir::GetNameFromLoc(val.getLoc()); if (!name_from_loc.empty()) return name_from_loc;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0)