Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 17 of 17 for try_emplace (0.22 sec)

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

        const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var,
        llvm::StringMap<PartitionedCallStackOpsInfo>*
            decomposed_partitioned_call_callees) {
      auto emplace_res = decomposed_partitioned_call_callees->try_emplace(
          callee.getName(), PartitionedCallStackOpsInfo());
      auto& info = emplace_res.first->second;
      // Recreate the call op with info.
      auto recreate_caller = [&] {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

        bool use_stablehlo_constant) {
      // Populate from metadata.
      ControlNodes control_nodes;
      for (const auto [from, to] : control_edges) {
        control_nodes.try_emplace(from);
        control_nodes[to].incoming.insert(from);
      }
    
      llvm::SmallVector<mlir::Type, 2> ret_types;
      llvm::SmallVector<mlir::Type, 4> input_types;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

          // Unsupported character in the equation.
          return std::nullopt;
        }
        if (map.count(equation[i])) {
          // Duplicate character in the equation.
          return std::nullopt;
        }
        map.try_emplace(equation[i], i);
      }
      return map;
    }
    
    std::optional<llvm::SetVector<char>> GetAvailableLabels(
        llvm::StringRef lhs, llvm::StringRef rhs, int* lhs_named_label_count,
        int* rhs_named_label_count) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size,
        llvm::StringMap<PartitionedCallDecompositionInfo>*
            decomposed_partitioned_call_callees) {
      auto emplace_res = decomposed_partitioned_call_callees->try_emplace(
          callee.getName(), PartitionedCallDecompositionInfo());
      auto& info = emplace_res.first->second;
      // Recreates the call op with info.
      auto recreate_caller = [&] {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        bool vars_initialized,
        llvm::SmallDenseMap<llvm::StringRef, PartitionedCallLiftingInfo>*
            lifted_callees) {
      auto emplace_res = lifted_callees->try_emplace(callee.getName(),
                                                     PartitionedCallLiftingInfo());
      if (emplace_res.second) {
        // Unseen callee. Perform resource lifting on it.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          }
        }
        if (auto tfl_operator =
                BuildOperator(real_inst, operands, results, intermediates)) {
          operation_index_to_operator_index.try_emplace(operation_index,
                                                        operators.size());
          operators.push_back(*tfl_operator);
          operators_in_mlir.push_back(real_inst);
        } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

          return errors::InvalidArgument(
              "Only feed output tensors of single output nodes are supported");
    
        // Collect mapping of OutputTensor to associated block arg.
        arg_nodes_to_values.try_emplace({arg_node.node, arg_node.index}, arg_def);
        island->getResult(0).replaceAllUsesWith(arg_def);
        // Erase control outputs from feed.
        auto control_uses = island->getResult(1).getUses();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top