- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 263 for created (0.15 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/unfreeze_constants.cc
pm.addPass(mlir::quant::CreateUnfreezeConstantsPass()); }, ctx, module_op)); if (const absl::Status create_dir_status = Env::Default()->CreateDir(std::string(checkpoint_dir)); !create_dir_status.ok()) { LOG(ERROR) << "Failed to create checkpoint directory at: " << checkpoint_dir; return create_dir_status; } TF_ASSIGN_OR_RETURN(const auto unused_variable_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 11:07:03 UTC 2024 - 3K bytes - Viewed (0) -
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/c/eager/c_api_test.cc
// only worker 1. // 2. Create a context B using A. // 3. Create the variable in B. // 4. Create another single host server def C with only worker 0. // 5. Start the GRPC server for worker 0 using C. // 6. Create a context D using the full cluster server def E. // 7. Read the variable in D. TEST(CAPI, SingleHostServerDefV1Works) { // Create a server def that represents a 2-process cluster and a client.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
#include "tensorflow/core/platform/types.h" struct MyCustomKernel { bool created; bool compute_called; }; static bool delete_called = false; static bool async_kernel_done = false; static void* MyCreateFunc(TF_OpKernelConstruction* ctx) { struct MyCustomKernel* s = new struct MyCustomKernel; s->created = true; s->compute_called = false; // Exercise attribute reads. TF_DataType type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K 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)