- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 230 for eraseOp (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
scalar_tensor.scalar<float>()() = 0.0f; auto flow = builder.create<TF::ConstOp>( ta.getLoc(), tensorflow::ConvertTensor(scalar_tensor, &builder).value()); ta.getFlow().replaceAllUsesWith(flow); ta.erase(); (*stats)[local_var].accumulate_on_write = false; return success(); } LogicalResult HandleTensorArrayReadV3Op( TF::TensorArrayReadV3Op read,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
} // Relocate the main function's FetchOp to the last. main_body.push_back(main_fetch_op->clone(mapper)); main_fetch_op.erase(); auto cloned_fetch_op = cast<FetchOp>(src_graph_op.GetFetch()->clone(mapper)); Value control_fetch = *cloned_fetch_op.getFetches().begin(); cloned_fetch_op.erase(); return control_fetch; } // Creates a new `IdentityOp` wrapped by an `IslandOp`. The identity op returns
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/tensorflow/transforms/decompose_reduce_dataset.cc
// Updates usage and erases rewritten reduce_dataset op based on the number // of state variables. for (int i = 0; i < state_size; ++i) { reduce_dataset.getResult(i).replaceAllUsesWith( dataset_while.getResult(i + 1)); } reduce_dataset.erase(); return WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
it.first->eraseOperands(it.second); } func.eraseArguments(args_to_erase); } // Erase all global tensors that were frozen. for (auto global_tensor : frozen_global_tensors) { remaining_global_tensor_ops.erase(global_tensor); global_tensor->erase(); } // Verify that there are no remaining global tensors. if (!allow_mutable_tensors && !remaining_global_tensor_ops.empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
if (failed(LiftHashTableOpsToArguments(module_op, caller_func))) { return failure(); } } // Erase the lifted operations explicitly. for (auto [lifted_op, arg_idx] : lifted_op_and_arg_idx) { lifted_op->erase(); } return success(); } void LiftHashTableOpsAsArgsPass::runOnOperation() { auto module_op = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/split_into_island_per_op_pass.cc
for (const auto& fetch : llvm::enumerate(graph_op.GetFetch().getFetches())) { if (fetch.value() == island_op.getControl()) { graph_op.GetFetch().getFetchesMutable().erase(fetch.index(), 1); break; } } island_op.erase(); } std::unique_ptr<OperationPass<func::FuncOp>> CreateSplitIntoIslandPerOpPass() { return std::make_unique<TF::SplitIntoIslandPerOpPass>(); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
assert(replicated_input->use_empty()); replicated_input->erase(); return WalkResult::advance(); }); if (result.wasInterrupted()) { signalPassFailure(); return; } getOperation()->walk([](TF::TPUPartitionedInputV2Op partitioned_input) { if (partitioned_input->use_empty()) partitioned_input->erase(); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0) -
staging/src/k8s.io/apiserver/pkg/admission/plugin/policy/internal/generic/informer.go
) var _ Informer[runtime.Object] = informer[runtime.Object]{} type informer[T runtime.Object] struct { cache.SharedIndexInformer lister[T] } // Creates a generic informer around a type-erased cache.SharedIndexInformer // It is incumbent on the caller to ensure that the generic type argument is // consistent with the type of the objects stored inside the SharedIndexInformer // as they will be casted.
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Feb 12 18:58:24 UTC 2024 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
// function, bail out. if (SymbolTable::lookupSymbolIn(module, name)) { module->emitWarning("session_initializer function already exists"); session_initializer.erase(); return; } auto init = builder.create<func::FuncOp>( module.getLoc(), name, FunctionType::get(module.getContext(), /*inputs=*/{}, /*results=*/{})); // Make savedmodel verification happy.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
rewriter.create<OutfeedOp>(op.getLoc(), token_type, op.getInputs(), token, /*outfeed_config=*/rewriter.getStringAttr("")); rewriter.eraseOp(op); return success(); } }; // Converts tf.TopKV2 to chlo.top_k. class ConvertTopKV2Op : public OpRewritePattern<TF::TopKV2Op> { public: using OpRewritePattern::OpRewritePattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)