Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for input_edges (0.22 sec)

  1. tensorflow/compiler/jit/build_xla_ops_pass.cc

      std::vector<const Edge*> input_edges_vector;
      TF_RETURN_IF_ERROR(n->input_edges(&input_edges_vector));
      absl::Span<const Edge*> input_edges(input_edges_vector);
    
      absl::c_transform(input_edges.subspan(0, num_constant_inputs),
                        std::back_inserter(result->constant_inputs),
                        IncomingEdgeAsOutput);
    
      absl::c_transform(
          input_edges.subspan(num_constant_inputs, num_non_constant_inputs),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

        TF_ASSIGN_OR_RETURN(const XlaFunctionInfo xla_function_info,
                            get_xla_function_info(*launch));
    
        std::vector<const Edge*> in_edges;
        TF_RETURN_IF_ERROR(launch->input_edges(&in_edges));
    
        const int num_inputs = in_edges.size();
        const int variable_start_index = xla_function_info.variable_start_index;
        const int num_variables = num_inputs - variable_start_index;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_test.cc

        TFE_TensorHandle* input_eager = TestScalarTensorHandle(eager_ctx, 2.0f);
        func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s));
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        input_eager = TestScalarTensorHandle(eager_ctx, 3.0f);
        func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s));
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 19 21:44:52 UTC 2023
    - 39.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

      const int kSliceSizeIndex = 2;
    
      const Edge* slice_input_edge;
      TF_RETURN_IF_ERROR(slice->input_edge(kSliceInputIndex, &slice_input_edge));
      const Edge* slice_size_edge;
      TF_RETURN_IF_ERROR(slice->input_edge(kSliceSizeIndex, &slice_size_edge));
      const Edge* slice_begin_edge;
      TF_RETURN_IF_ERROR(slice->input_edge(kSliceBeginIndex, &slice_begin_edge));
    
      SliceInputs slice_inputs;
      slice_inputs.input =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/encapsulate_util.cc

      std::map<std::pair<string, int>, Node*> placeholders;
      for (int i = 0, end = edges.size(); i < end; i++) {
        Node* dst = g->FindNodeId(edges[i].dst_node_id);
        const Edge* e;
        TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e));
        Node* src = e->src();
        int src_output = e->src_output(), dst_input = e->dst_input();
        g->RemoveEdge(e);
    
        // Find or create placeholder node.
        string new_name =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/node_matchers.cc

        }
      }
    
      bool MatchAndExplainInput(const Node* node, int input_idx,
                                ::testing::MatchResultListener* listener) const {
        const Edge* edge;
        if (!node->input_edge(input_idx, &edge).ok()) {
          if (listener->IsInterested()) {
            *listener << "\ncould not find incoming edge for input " << input_idx;
          }
          return false;
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 03 16:15:20 UTC 2022
    - 16.8K bytes
    - Viewed (0)
Back to top