Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 382 for get_operations (0.34 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py

      """
      output_renaming_map = {}
      with session.Session(graph=ops.Graph()):
        importer.import_graph_def(graph_def, name='')
        graph = ops.get_default_graph()
        for op in graph.get_operations():
          if op.type == '_Retval':
            expected_node_name = op.name
            if op.get_attr('tf_saved_model.index_path') is not None:
              index_path_name = op.get_attr('tf_saved_model.index_path')[0]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_executor_to_functional.cc

                    "expected tf_executor.island";
    
        // Move inner ops in island to before the outer graph.
        auto& island_body = island_op.GetBody().getOperations();
        parent_block->getOperations().splice(graph_position, island_body,
                                             island_body.begin(),
                                             std::prev(island_body.end()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc

      for (Block& block : region->getBlocks()) {
        for (Operation& op : block.getOperations()) {
          if (op.hasTrait<OpTrait::ReturnLike>()) result = &op;
        }
      }
      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;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

      // Move all inner ops of the launch to the block containing the launch.
      auto body = launch.GetBody().without_terminator();
      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() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc

      for (func::FuncOp func_op :
           llvm::make_early_inc_range(nested_block.getOps<func::FuncOp>())) {
        if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) {
          nested_block.getOperations().remove(func_op.getOperation());
          symbol_table.insert(func_op.getOperation());
        }
      }
      nested_module->erase();
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 08:06:04 UTC 2023
    - 4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc

                     const TF::SideEffectAnalysis::Info& side_effect_analysis) {
      // IfRegion ops must be in the same block.
      if (second_if.getOperation()->getBlock() !=
          first_if.getOperation()->getBlock()) {
        return false;
      }
      assert(first_if.getOperation()->isBeforeInBlock(second_if.getOperation()));
    
      llvm::SmallSetVector<Operation*, 4> destination_ops =
          GetAllOpsFromIf(first_if);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 25.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tf_functional_to_executor.cc

        to_fetch = to_fetch.drop_back();
      }
      builder.create<tf_executor::FetchOp>(loc, to_fetch);
      // Build Island.
      island.getBody().push_back(new Block);
      island.getBody().front().getOperations().splice(
          island.getBody().front().begin(), body.getOperations(),
          copy_range.begin(), copy_range.end());
      builder.setInsertionPointToEnd(&island.getBody().front());
      builder.create<tf_executor::YieldOp>(loc, return_op.getOperands());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc

            }
            if (constant.hasOneUse()) {
              LLVM_DEBUG(llvm::dbgs() << "Moved constant " << constant << "\n");
              const_op.getOperation()->moveBefore(&body.begin()->front());
              return;
            }
            map_entry.first->getSecond() = const_op.clone();
            body.begin()->getOperations().insert(body.begin()->begin(),
                                                 map_entry.first->getSecond());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc

             llvm::zip(block.getArguments(), op->getOperands())) {
          std::get<0>(arg_mapping).replaceAllUsesWith(std::get<1>(arg_mapping));
        }
        op->getBlock()->getOperations().splice(op->getIterator(),
                                               block.getOperations(), block.begin(),
                                               std::prev(block.end()));
        Operation &return_op = block.front();
        for (auto ret_mapping :
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

        replaceAllUsesInRegionWith(std::get<0>(p), std::get<1>(p), op_region);
      }
    
      // Move all instructions in cluster_op into outlined_function's only block.
      auto& op_body = op.GetBody().getOperations();
      outlined_func_block->getOperations().splice(
          outlined_func_block->end(), op_body, op_body.begin(), op_body.end());
    
      // Replace `tf_device.return` terminator with `std.return` in function
      // body.
      auto return_op =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
Back to top