Search Options

Results per page
Sort
Preferred Languages
Advance

Results 111 - 120 of 1,997 for Reserve (0.15 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc

    template <typename ValueT, typename... RangeTs>
    llvm::SmallVector<ValueT, 4> Concat(RangeTs&&... ranges) {
      llvm::SmallVector<int64_t, 4> results;
      results.reserve(Size(std::forward<RangeTs>(ranges)...));
      Append(results, std::forward<RangeTs>(ranges)...);
      return results;
    }
    
    // A struct to hold information about dimensions of dot_general operands.
    class DotDimensionsInfo {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/get_compiler_ir.cc

      TF_RET_CHECK(fbody != nullptr);
      auto& input_args = fbody->record->fdef().signature().input_arg();
      int input_arg_size = input_args.size();
      std::vector<XlaCompiler::Argument> args;
      args.reserve(input_arg_size);
    
      for (auto& arg_info : flat_arg_shape_and_dtype) {
        XlaCompiler::Argument arg;
        arg.kind = XlaCompiler::Argument::kParameter;
        arg.type = arg_info.dtype;
        arg.shape = arg_info.shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc

        tf_executor::IslandOp island_op, tf_device::ReplicateOp replicate_op,
        int num_replicas, llvm::SmallVectorImpl<tf_executor::IslandOp>& replicas,
        bool legacy_graph_export, int replica_group_idx) {
      replicas.reserve(num_replicas);
      auto devices = replicate_op.getDevices();
    
      // Collect result types and operands.
      Operation& terminator = replicate_op.GetBody().back();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 24 21:01:40 UTC 2023
    - 16.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc

        const auto predecessors =
            side_effect_analysis.DirectControlPredecessors(&sub_op);
        // Get the controls from the predecessors.
        llvm::SmallVector<Value, 4> predecessor_controls;
        predecessor_controls.reserve(predecessors.size());
        for (auto predecessor : predecessors) {
          predecessor_controls.push_back(new_control_for_sub_ops[predecessor]);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 16.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

      for (int i = 0; i < inputs.size(); i++) {
        input_ids[i] = ToId(inputs[i]);
        input_dtypes[i] = inputs[i]->DataType();
      }
      std::vector<TapeTensor> tape_tensors;
      tape_tensors.reserve(outputs.size());
      for (auto t : outputs) {
        tape_tensors.push_back(TapeTensor(t));
      }
      GradientTape::RecordOperation(
          op_name, tape_tensors, input_ids, input_dtypes,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  6. src/runtime/sys_freebsd_arm.s

    	MOVW new+4(FP), R1		// arg 2 act
    	MOVW old+8(FP), R2		// arg 3 oact
    	MOVW $SYS_sigaction, R7
    	SWI $0
    	MOVW.CS	$-1, R0
    	MOVW	R0, ret+12(FP)
    	RET
    
    TEXT runtimeĀ·sigtramp(SB),NOSPLIT|TOPFRAME,$0
    	// Reserve space for callee-save registers and arguments.
    	MOVM.DB.W [R4-R11], (R13)
    	SUB	$16, R13
    
    	// this might be called in external code context,
    	// where g is not set.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Jun 06 18:49:01 UTC 2023
    - 10.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc

          // Cast values
          std::vector<Eigen::half> new_values;
          DenseFPElementsAttr value_attr =
              mlir::cast<DenseFPElementsAttr>(op.getValue());
          new_values.reserve(value_attr.getNumElements());
    
          constexpr float kMaxFloat16Value = 65504.f;
          constexpr float kMinFloat16Value = -65504.f;
    
          for (auto value : value_attr.template getValues<float>()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc

      llvm::SmallVector<Operation*, 4> result;
      auto it = sorted_control_predecessors_.find(op);
      if (it == sorted_control_predecessors_.end()) return result;
      result.reserve(it->getSecond().size());
      for (auto predecessor : it->getSecond()) {
        if (!filter || filter(predecessor)) result.push_back(predecessor);
      }
      return result;
    }
    
    const llvm::SmallVector<Operation*, 4>&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 41.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc

      argument_kinds.clear();
      if (input_types_str.empty()) return absl::OkStatus();
    
      std::vector<absl::string_view> argument_kind_strs =
          absl::StrSplit(input_types_str, ',');
      argument_kinds.reserve(argument_kind_strs.size());
      for (const auto& argument_kind_str : llvm::enumerate(argument_kind_strs)) {
        const auto& value = argument_kind_str.value();
        if (value == "parameter") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.8K bytes
    - Viewed (0)
  10. tensorflow/c/c_test_util.cc

      return grads;
    }
    
    std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) {
      std::vector<string> names;
      auto functions = graph_def.library().function();
      names.reserve(functions.size());
      for (const tensorflow::FunctionDef& func : functions) {
        names.push_back(func.signature().name());
      }
      std::sort(names.begin(), names.end());
      return names;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
Back to top