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

  /external/tensorflow/tensorflow/core/kernels/
batch_kernels.cc 334 const int num_input_edges = batch->task(0).inputs.size(); local
337 for (int i = 0; i < num_input_edges; ++i) {
416 task.context->allocate_output(num_input_edges, index_shape, &output),
424 task.context->allocate_output(num_input_edges + 1,
431 EmitIndexTensor(last_task_context, *batch, num_input_edges),
    [all...]
  /external/tensorflow/tensorflow/core/graph/
graph_partition.cc 997 int32 num_input_edges = 0; local
    [all...]

Completed in 361 milliseconds