Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for fetch_op (0.26 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc

      GraphOp graph_op = GetGraphOpFromFuncOp(init_func_op);
      if (!graph_op) return success();  // Consider empty FuncOp valid.
    
      FetchOp fetch_op = graph_op.GetFetch();
      for (const Value fetch : fetch_op.getFetches()) {
        if (!mlir::isa<tf_executor::ControlType>(fetch.getType())) {
          fetch_op.emitError(absl::StrFormat(
              "Validation failed for the initializer function: %s. "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:54:52 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc

        if (!HasSingleOpInBlock<IslandOp>(&block)) return failure();
    
        FetchOp fetch_op = op.GetFetch();
        auto island_op = llvm::cast<IslandOp>(block.front());
        YieldOp yield_op = island_op.GetYield();
    
        // Map graph results to inner ops results of single island.
        llvm::SmallVector<Value, 8> new_rets;
        for (Value operand : fetch_op.getFetches()) {
          // Control results should not be propagated out.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 42.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc

    // `FetchOp` will be used without modified. Returns the fetch operands from the
    // `scr_graph_op`.
    Value CloneGraphOps(GraphOp src_graph_op, GraphOp dst_graph_op,
                        IRMapping& mapper) {
      Block& main_body = dst_graph_op.GetBody();
    
      // Take the reference of the main graph's FetchOp to later move to the end.
      FetchOp main_fetch_op = dst_graph_op.GetFetch();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      args_[arg] = node;
      return absl::OkStatus();
    }
    
    // Creates return nodes per operand of a FetchOp. If names is supplied, those
    // names will be used per node in order instead of generating a unique name.
    Status Exporter::AddFetchNode(FuncOp function, mlir::tf_executor::FetchOp fetch,
                                  llvm::ArrayRef<llvm::StringRef> names) {
      auto& return_nodes = returns_[fetch];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      args_[arg] = node;
      return absl::OkStatus();
    }
    
    // Creates return nodes per operand of a FetchOp. If names is supplied, those
    // names will be used per node in order instead of generating a unique name.
    Status Exporter::AddFetchNode(FuncOp function, mlir::tf_executor::FetchOp fetch,
                                  llvm::ArrayRef<llvm::StringRef> names) {
      auto& return_nodes = returns_[fetch];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

                                    executor_graph.getBody().end());
    
        // Add terminator of tf_executor::graph
        rewriter.setInsertionPointToEnd(&executor_graph.getBody().front());
        rewriter.create<tf_executor::FetchOp>(loc);
    
        // Add terminator of func
        rewriter.setInsertionPointToEnd(&func.getBody().front());
        rewriter.create<func::ReturnOp>(loc);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc

        if (!MayHaveSideEffect(op)) return;
        // Skip following ops to avoid that every island, graph and function is
        // classified as unknown side-effecting.
        if (isa<tf_executor::YieldOp, tf_executor::FetchOp,
                mlir::func::ReturnOp>(op))
          return;
    
        // Propagate side effects from regions or functions attached to `op` for
        // some special cases.
        if (auto func = llvm::dyn_cast<func::FuncOp>(op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 41.2K bytes
    - Viewed (0)
  8. pkg/wasm/cache_test.go

    	}
    	manifest.MediaType = types.DockerManifestSchema2
    
    	// Push image to the registry.
    	err = crane.Push(img, ref, fetchOpt)
    	if err != nil {
    		t.Fatal(err)
    	}
    
    	// Push image to the registry with latest tag as well
    	ref = fmt.Sprintf("%s/test/valid/docker:latest", host)
    	err = crane.Push(img, ref, fetchOpt)
    	if err != nil {
    		t.Fatal(err)
    	}
    
    	// Calculate sum
    	d, _ := img.Digest()
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Wed Apr 17 20:06:41 UTC 2024
    - 33.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

            updated_next_iteration = true;
          }
        }
      } while (updated_next_iteration);
    }
    
    // Propagates devices to function results.
    void PropagateDevicesToResults(
        func::FuncOp func, tf_executor::FetchOp fetch,
        const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) {
      // We apply all result attributes at once to avoid excessive allocations when
      // we have many result values.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc

      return control_island.getOutputs().empty() &&
             isa<TF::NoOp>(control_island.GetBody().front());
    }
    
    // Remove all control outputs of the function. Traverses NoOp control barrier
    // chains from FetchOp to all NoOp control barriers. Returns true
    // iff at least one control output is deleted. The ops_connected_to_fetch
    // set is populated with all operations that had a direct (or indirect, through
    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