HomeSort by relevance Sort by last modified time
    Searched defs:input_edge (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
shape_refiner.cc 420 const Edge* input_edge; local
421 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge));
422 OutputTensor tensor(input_edge->src(), input_edge->src_output());
454 const Edge* input_edge; local
455 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge));
457 InferenceContext* src_context = GetContext(input_edge->src());
459 ShapeHandle src_shape = src_context->output(input_edge->src_output())
    [all...]
  /external/tensorflow/tensorflow/core/graph/
graph.cc 238 Status Node::input_edge(int idx, const Edge** e) const { function in class:tensorflow::Node
240 return errors::InvalidArgument("Invalid input_edge index: ", idx, ", Node ",
291 TF_RETURN_IF_ERROR(input_edge(idx, &e));
309 TF_RETURN_IF_ERROR(input_edge(idx, &e));

Completed in 361 milliseconds