Search Options

Results per page
Sort
Preferred Languages
Advance

Results 141 - 150 of 234 for eraseOp (0.1 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc

      auto new_op = builder->create<OpT>(op->getLoc(), op->getResultTypes(),
                                         operands, attr.getAttrs());
      op->replaceAllUsesWith(new_op.getOperation()->getResults());
      op->erase();
      return new_op;
    }
    
    // Returns success if the function has at most one op of the template type and
    // assigns it to `result`, if present. If there are multiple such ops, returns
    // failure.
    template <typename OpT>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 22:55:42 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

            TransposeOp transpose = tranpose_op;
            transpose.getOperation()->moveBefore(op);
            transpose.setOperand(0, operand->get());
            transpose.setOperand(1, permutation_op);
            transpose_ops->erase(it);
            return transpose;
          }
        }
      }
      return nullptr;
    }
    
    #define GEN_PASS_DEF_LAYOUTASSIGNMENTPASS
    #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc

              LLVM_DEBUG(llvm::dbgs() << "Re-use sunk constant " << use->get()
                                      << "\n     in " << use->get() << "\n");
              if (constant.use_empty()) const_op.erase();
              return;
            }
            if (constant.hasOneUse()) {
              LLVM_DEBUG(llvm::dbgs() << "Moved constant " << constant << "\n");
              const_op.getOperation()->moveBefore(&body.begin()->front());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc

        if (!best) {
          assert(ready.empty());
          return result;  // happens for unused results for ops in the todo list
        }
    
        // Consider this operation emitted, and make its results available.
        ready.erase(std::find(ready.begin(), ready.end(), best));
        previous_op = best;
        for (Value result : best->getResults()) {
          todo.push(result);
        }
        for (Operation* successor : ctrlSuccessors(best)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 08 17:01:11 UTC 2022
    - 5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc

        for (auto [result, new_result] :
             llvm::zip_equal(op.getResults(), new_op.getResults())) {
          new_op_mapper.map(result, new_result);
        }
      }
    
      call_op.erase();
    }
    
    void UnwrapXlaCallModuleOpPass::runOnOperation() {
      ModuleOp module_op = getOperation();
      SymbolTable symbol_table(module_op);
    
      for (auto func_op : module_op.getOps<func::FuncOp>()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/cc/static_range_ptq.cc

                              *ctx, *module));
    
      // Remove the `tpu` tag for exporting because the output quantized model is
      // essentially a CPU model.
      tags.erase("tpu");
    
      py_function_library.SaveExportedModel(
          dst_saved_model_path, post_calibrated_exported_model,
          src_saved_model_path, tags, signature_def_map);
    
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc

          return success();
        }
    
        // Remove TPU operations.
        if (isa<TF::TPUReplicateMetadataOp, TF::TPUCompilationResultOp,
                TF::TPUOrdinalSelectorOp>(op)) {
          op->erase();
        } else if (auto replicated_input_op =
                       dyn_cast_or_null<TF::TPUReplicatedInputOp>(op)) {
          // TODO(b/267700110): Handle multiple input/output cases.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

      Operation* launch_op = launch.getOperation();
      launch_op->getBlock()->getOperations().splice(
          launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(),
          body.end());
    
      launch.erase();
    
      return success();
    }
    
    void LaunchToDeviceAttributePass::runOnOperation() {
      const Dialect* tf_dialect = getContext().getLoadedDialect("tf");
      if (!tf_dialect) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc

          block.addArgument(read_operand.getType(), loc);
    
        func.setType(FunctionType::get(&getContext(), block.getArgumentTypes(),
                                       func.getResultTypes()));
        cluster_func.erase();
      }
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreateTPUResourceReadForWritePass() {
      return std::make_unique<TPUResourceReadForWritePass>();
    }
    
    }  // namespace TFTPU
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 16:54:40 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  10. src/internal/chacha8rand/chacha8.go

    	if s.c == ctrMax-ctrInc {
    		s.n = uint32(len(s.buf)) - reseed
    	}
    }
    
    // Reseed reseeds the state with new random values.
    // After a call to Reseed, any previously returned random values
    // have been erased from the memory of the state and cannot be
    // recovered.
    func (s *State) Reseed() {
    	var seed [4]uint64
    	for i := range seed {
    		for {
    			x, ok := s.Next()
    			if ok {
    				seed[i] = x
    				break
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon May 13 21:47:29 UTC 2024
    - 4.4K bytes
    - Viewed (0)
Back to top