Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for tf_outputs (0.16 sec)

  1. tensorflow/c/c_api.cc

      cond_graph->parent_inputs = inputs;
      body_graph->parent = g;
      body_graph->parent_inputs = inputs;
    
      TF_Output* cond_inputs = new TF_Output[ninputs];
      TF_Output cond_output = {nullptr, -1};
      TF_Output* body_inputs = new TF_Output[ninputs];
      TF_Output* body_outputs = new TF_Output[ninputs];
      for (int i = 0; i < ninputs; ++i) body_outputs[i] = {nullptr, -1};
      const char* name = nullptr;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"),
                               "_XlaRecvAtHost", opts.op_registry());
      node_builder.Input(std::move(key_input));
      return opts.WithAttr("Toutputs", dtypes)
          .WithAttr("key", key)
          .WithAttr("device_ordinal", 0)
          .WithAttr("_encapsulate", cluster)
          .WithAttr("_outside", oc_cluster)
          .FinalizeBuilder(&node_builder);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

      NodeDefBuilder recv_at_host_builder(
          absl::StrCat("outside_compilation_", oc_cluster_name, "_recv"),
          "_XlaRecvAtHost");
      NodeDef recv_at_host_def;
      recv_at_host_builder.Attr("Toutputs", recv_at_host_dtypes);
      // The correct device_ordinal will be inserted during replication in a
      // subsequent rewrite.
      AttrValue device_ordinal_value;
      device_ordinal_value.set_placeholder("_device_ordinal");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      "tf._SomeOp"(%0#1, %0#0) : (tensor<!tf_type.resource>, tensor<f32>) -> ()
      // CHECK: "tf._SomeOp"(%[[ARG2]], %[[IF_OUTPUT]]) : (tensor<!tf_type.resource>, tensor<f32>) -> ()
      "tf._SomeOp"(%0#3, %0#2) : (tensor<!tf_type.resource>, tensor<f32>) -> ()
      // CHECK: return %[[IF_OUTPUT]] : tensor<f32>
      func.return %0#2 : tensor<f32>
    }
    
    // CHECK-LABEL: func @eliminatePassThroughCaseRegion(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir

        // CHECK:            "tf._XlaSendFromHostV2"(%[[E_OUTPUT]], %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]])
        // CHECK:            %[[F_OUTPUT:[0-9]*]] = "tf.F"(%[[C_OUTPUT]]#1)
        // CHECK:            "tf._XlaSendFromHostV2"(%[[F_OUTPUT]], %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]])
        // CHECK:            %[[RECV_OUTPUT2:[0-9]*]]:2 = "tf._XlaRecvAtHostV2"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 129.6K bytes
    - Viewed (0)
Back to top