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

  /external/tensorflow/tensorflow/core/kernels/
redux_functor.h 50 const int64 max_parallelism = total_workload / parallel_cell_size; local
56 std::min(max_parallelism, Eigen::divup(total_workload, min_block_size));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.cc 32 SimpleCostModel(const int64 max_parallelism,
34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {}
53 DefaultCostModel(const int64 max_parallelism,
56 : max_parallelism_(max_parallelism),
65 int64 max_parallelism; variable
77 max_parallelism =
84 max_parallelism = max_parallelism_;
97 return std::min(max_parallelism,
108 const int64 max_parallelism,
112 VLOG(1) << "ParallelTaskAssignment max_parallelism: " << max_parallelism
    [all...]
cpu_compiler.cc 368 const int max_parallelism = local
379 max_parallelism, ShapeSizeBytesFunction(), target_machine_features);
    [all...]
  /external/tensorflow/tensorflow/core/kernels/data/experimental/
threadpool_dataset_op.cc 310 auto max_parallelism = dataset()->max_intra_op_parallelism_; variable
312 RunnerWithMaxParallelism(*ctx->runner(), max_parallelism);

Completed in 639 milliseconds