- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 93 for created (0.1 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/quantize_passes.cc
mlir::quant::stablehlo::FillPresetQuantizationOptions( quantization_options); } // TODO: b/276999414 - Add activation and bias quantization component as // respective quantization passes are created. QuantizationComponentSpec weight_component; for (const auto& component : quantization_options_.quantization_method() .custom_quantization_method()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
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()); new_launch_op.GetBody().getTerminator()->setOperands(new_launch_op_results); return new_launch_op; } // Create the new device launch op which wraps the copy op.
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/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
new_results))) return failure(); // Update the regions. The dialect conversion framework wants new regions to // 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,
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/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
OpBuilder replacer(yield_op); island_op.getBody().push_back(new Block); replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. {
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/jit/pjrt_base_device.cc
options.shape_determination_fns) { if (options.shape_determination_fns.empty()) { LOG(ERROR) << "shape_representation_fns must be non-empty."; } VLOG(1) << "Created PJRT base device " << options.compilation_device_name << " device_name: " << name(); } /*static*/ absl::StatusOr<const PjRtBaseDevice::Metadata*> PjRtBaseDevice::GetMetadataFromDevice(DeviceBase* device) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
// Set insertion point to right after all operations in cluster. builder->setInsertionPoint(c.ops.back()->getNextNode()); // Create a stand-alone region to hold all instructions in the cluster. Region region; region.push_back(new Block); // Move all operations in cluster to newly created region, stripping their // "device" attribute since launch op already carries device information. Block* block = ®ion.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
class TPUAnnotateDynamicShapeInputsPass : public impl::TPUAnnotateDynamicShapeInputsPassBase< TPUAnnotateDynamicShapeInputsPass> { void runOnOperation() override; }; // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
return isa<NameLoc>(loc) || (isa<CallSiteLoc>(loc) && isa<NameLoc>(mlir::cast<CallSiteLoc>(loc).getCallee())); }); } // Finds the pattern of the location created by `ImporterBase::GetLocation` // in `tensorflow/compiler/mlir/tensorflow/translate/import_model.cc`. void FindQuantizationUnitsRecursively(Location loc, SmallVector<QuantizationUnit>& units) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Create tensor attribute for this variable. absl::StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); if (!tensor_attr_or.ok()) { return module.emitOpError() << "failed to convert tensor (name: " << name.str() << ")"; } ElementsAttr tensor_attr = tensor_attr_or.value(); builder.create<tf_saved_model::GlobalTensorOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
// Finally, use the quantization parameter to create the quantize and // dequantize ops, and insert them between the tf.FakeQuantWithMinMaxVarsOp // and its users. Value value = tf_op.getOutputs(); auto quantize = rewriter.create<quantfork::QuantizeCastOp>( tf_op.getLoc(), qtype.getValue(), value); auto dequantize = rewriter.create<quantfork::DequantizeCastOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0)