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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.h 67 : max_parallelism_(max_parallelism), shape_size_function_(shape_size) {}
96 HloCostAnalysis::ShapeSizeFunction shape_size_function_; member in class:xla::cpu::ParallelTaskAssigner
parallel_task_assignment.cc 233 shape_size_function_, module);

Completed in 203 milliseconds