Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 234 for eraseOp (0.12 sec)

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

      func_op.eraseArguments(args_to_erase_bit_vector);
    
      // Erases entries in "tf._input_shapes" attribute of `func_op` that correspond
      // to the erased arguments.
      if (auto input_shapes_attr =
              func_op->getAttrOfType<ArrayAttr>(kTfInputShapesAttr);
          input_shapes_attr) {
        // Construct a new array of input shapes excluding the input shapes of the
        // erased arguments.
        SmallVector<Attribute> updated_input_shapes_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

                    "of indices");
        }
      }
    
      if ((indices_rank == 0) || (indices_rank == batch_dims_i)) {
        // Scalar indices (output is rank(params) - 1).
        // Erase shape[axis]
        shape.erase(shape.begin() + axis_i);
      } else if (indices_rank == 1) {
        // Vector indices (output is rank(params)).
        // Copy indices.shape into params.shape[axis]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

        for (auto it :
             llvm::zip(control_node_op.getOutputs(), cloned_inner->getResults())) {
          std::get<0>(it).replaceAllUsesWith(std::get<1>(it));
        }
        rewriter.eraseOp(outer_op);
      }
      return control_edges;
    }
    
    }  // namespace
    
    namespace tflite {
    
    bool MlirToFlatBufferTranslateFunction(mlir::ModuleOp module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc

        }
      }
      return result;
    }
    
    void EraseReturnOperands(Region* region, llvm::BitVector& erase) {
      for (Block& block : region->getBlocks()) {
        for (Operation& op : block.getOperations()) {
          if (!op.hasTrait<OpTrait::ReturnLike>()) continue;
          op.eraseOperands(erase);
        }
      }
    }
    
    // Erases the given results from an operation, similar to what
    // Operation::eraseArguments does (but for results).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc

      dead_ops.insert(vars.begin(), vars.end());
    
      // Removes relevant ops in topological order.
      for (auto& op : vars) RecursiveRemove(op, erase_list, dead_ops);
    
      // Erases the ops.
      for (auto op : erase_list) op->erase();
    }
    
    void RemoveVariablesInSessionInitializerPass::runOnOperation() {
      ModuleOp module_op = getOperation();
    
      for (auto init_func_op : GetInitializerFunctions(module_op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc

      }
    
      // Erase unreachable ops in reverse order so references don't need to be
      // dropped before removing an op. Going in reverse order will guarantee that
      // when an op to be erased is reached, there are no users left.
      for (Operation& op :
           llvm::make_early_inc_range(llvm::reverse(graph.GetBody())))
        if (!reachable_ops.contains(&op)) op.erase();
    }
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 7.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_optimize_global_tensors.mlir

      // argument is not erased.
    
      "tf_saved_model.global_tensor"() { sym_name = "exported_unbound", tf_saved_model.exported_names = ["exported_unbound"], type = tensor<f32>, value = dense<42.> : tensor<f32> } : () -> ()
      // CHECK: sym_name = "exported_unbound"
    
      // Test case: Check that a global tensor that isn't even bound to an argument
      // is erased.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:49:35 UTC 2023
    - 9.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc

          CreateRemoteRunCalls(context, *metadatas);
    
          // Erases the original operations which have been cloned in the remote
          // functions.
          for (auto &iter : *metadatas) {
            llvm::StringRef host = iter.first();
            FunctionMetadata &metadata = iter.second;
            // Do not erase operations placed on the localhost.
            if (IsOnLocalHost(host)) continue;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  9. subprojects/core/src/main/java/org/gradle/api/internal/tasks/options/MethodSignature.java

    import java.lang.invoke.MethodType;
    import java.lang.reflect.Method;
    import java.util.Objects;
    
    /**
     * A representation of a method signature. Contains the method name, erased parameter types and erased return type.
     */
    public final class MethodSignature {
        public static MethodSignature from(Method method) {
            return new MethodSignature(method.getName(), MethodType.methodType(
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed Feb 16 17:15:11 UTC 2022
    - 1.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc

      while (!control_barrier_worklist.empty()) {
        Value control_barrier = control_barrier_worklist.front();
        control_barrier_worklist.pop();
    
        // We can only erase control barriers whose uses have been erased as well.
        if (!control_barrier.use_empty()) continue;
    
        // Only values defined by IslandOp were inserted in the worklist.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.7K bytes
    - Viewed (0)
Back to top