- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 263 for created (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
return builder.create<TF::ConstOp>( loc, DenseStringElementsAttr::get( tensor_type, SmallVector<StringRef>(str_values.begin(), str_values.end()))); } // Creates a 1D string array constant for "tensor_names" input of `RestoreV2` // op. The `ConstOp` will be created at `builder`'s current insertion point.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop.cc
// before the NextIteration node is created. string NextIterationName(const Scope& scope, int loop_var_idx) { string result; const string& prefix = scope.impl()->name(); if (!prefix.empty()) strings::StrAppend(&result, prefix, "/"); strings::StrAppend(&result, "NextIteration"); if (loop_var_idx > 0) strings::StrAppend(&result, "_", loop_var_idx); return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
tf_op->emitError("malformed opdef"); return failure(); } if (tf_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>()) { // Add derived "operand_segment_sizes" attr to the created operation. // TODO(b/146937733): Don't use <void> here. set_segment_sizes_attr(input_ranges, op_def->input_arg(), mlir::OpTrait::AttrSizedOperandSegments<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
llvm::cl::init("")); // A list of comma separated TF operators which are created by the user. // This must be used with `-emit-select-tf-ops=true`. // NOLINTNEXTLINE opt<std::string> select_user_tf_ops( "select-user-tf-ops", llvm::cl::desc( "<list of custom tf ops created by the user (comma separated)>"), llvm::cl::init("")); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
fetches.emplace_back(fetch_operand); auto builder = OpBuilder::atBlockTerminator(&main_graph_op.GetBody()); builder.create<FetchOp>(main_graph_op.getLoc(), std::move(fetches)); } // Creates a new Location for the initializer function. This creates a loc by // attaching a to the initializer function's type so that it is identifiable. Location CreateInitOpLoc(MLIRContext* ctx, func::FuncOp init_func_ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
if (!buffer_or.ok()) { done(buffer_or.status()); return; } xla::PjRtBuffer* pjrt_buffer = (*buffer_or).get(); if (use_pjrt_tensor_buffer_) { // Copy the newly created tensor with PjRtTensorBuffer to output device // tensor. absl::StatusOr<Tensor> t = MakeTensorFromPjRtBuffer( device_tensor->dtype(), device_tensor->shape(), std::move(*buffer_or)); if (!t.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
func_input_arg = block->getArgument(0); return quantization_func; } // Post-actions after adding quantization logics. Post-actions include // 1) Adding the created function in the symbol table // 2) Creating a PartitionedCallOp in the main graph that calls the created // function. TF::PartitionedCallOp FinalizeFunctionRegister( PatternRewriter& rewriter, Value input, Value output,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
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/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/quantization/tensorflow/passes/unfreeze_constants.cc
// creates a new session_initializer op and returns it. SessionInitializerOp GetOrCreateSessionInitializerOp(ModuleOp module_op) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module_op); // Create one if it doesn't exist. if (!session_init_op) { OpBuilder builder(&module_op.getBodyRegion()); session_init_op = builder.create<SessionInitializerOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0)