- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 235 for created (0.09 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_weight.cc
// Create new ConstantOp-ConvertOp-Operation sequences. At this moment, // old ConstantOp is guaranteed to have one F32->F16 convert op regardless // of its number of users. rewriter.setInsertionPointAfter(op); // create new F16 constant op in that location ConstantOp new_const = rewriter.create<ConstantOp>(
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/ir/tf_ops_layout_helper.cc
return {0, 2, 3, 1}; } else { return {}; } } // Shuffle elements in the `attr` according to the permutation. Optional // `inner_size` allows to shuffle array attributes created from rank 2 tensors // on outer dimension only. ArrayAttr ShuffleArrayAttr(ArrayAttr attr, ArrayRef<int64_t> permutation, int inner_size) { if (attr.empty()) return attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc
} } }); } // Creates an instance of the default quant parameters pass. std::unique_ptr<OperationPass<func::FuncOp>> CreateImportQuantStatsPass( OperationToName op_to_name, const std::string &stats_str) { auto pass = std::make_unique<ImportQuantStatsPass>(op_to_name); if (pass->ParseQuantStats(stats_str)) return nullptr; return pass; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; bool HasOutsideCompilationAttribute(Operation* op) { return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } // 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 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
auto builder = OpBuilder::atBlockTerminator(&main_graph_op.GetBody()); // Create an IslandOp that will wrap the IdentityOp. Add a control dependency // for the newly copied save function. auto wrapper_island_op = builder.create<IslandOp>( name_loc, TypeRange{main_file_prefix_arg.getType()}, tf_executor::ControlType::get(&ctx), ValueRange(control_inputs));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/mlir_dump.cc
const std::string dump_file_path = tsl::io::JoinPath(*dump_dir, dump_file_name); TF_ASSIGN_OR_RETURN(std::unique_ptr<llvm::raw_ostream> file, WritableFileWrapper::Create(dump_file_path)); LOG(INFO) << "IR dump file created: " << dump_file_path; return file; } class PrinterConfig : public mlir::PassManager::IRPrinterConfig { public: explicit PrinterConfig(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 05:38:57 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
auto begin_op = rewriter.create<arith::ConstantOp>(op->getLoc(), type, begin_attr); auto end_attr = DenseElementsAttr::get<int32_t>(type, padded_end); auto end_op = rewriter.create<arith::ConstantOp>(op->getLoc(), type, end_attr); auto stride_attr = DenseElementsAttr::get<int32_t>(type, padded_stride); auto stride_op = rewriter.create<arith::ConstantOp>(op->getLoc(), type, stride_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/quantization_unit_loc.cc
// If the op is rewritten, FusedLoc can be created. for (Location child_loc : mlir::cast<FusedLoc>(loc).getLocations()) { std::optional<QuantizationUnitLoc::QuantizationUnit> found_unit = FindQuantizationUnitFromLoc(child_loc); if (found_unit.has_value()) return found_unit; } } else if (isa<CallSiteLoc>(loc)) { // If the graph is inlined, CallSiteLoc can be created. return FindQuantizationUnitFromLoc(
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/tensorflow/transforms/executor_island_coarsening.cc
OpBuilder builder(merged_island.insert_point); auto new_island = builder.create<IslandOp>( merged_island.insert_point->getLoc(), result_types, operands); new_island.getBody().push_back(new Block); return new_island; } // Creates respective YieldOp for the new merged island. YieldOp CreateNewIslandYieldOp(IslandOp new_island,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
const Type expressed_type = value.getType(); // The value needs to be requantized. A Quantize op will be created to use // it as the operand and replace its uses. const Type new_type = state.params.castFromExpressedType(expressed_type); if (!new_type) return; auto requantize_op = builder_.create<quantfork::QuantizeCastOp>(loc, new_type, value); value.replaceAllUsesWith(requantize_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0)