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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.h 195 // Decorates shape_size_ by returning 0 immediately if the shape does not have
202 const ShapeSizeFunction shape_size_; member in class:xla::HloCostAnalysis
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.cc 34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {}
39 const int64 instruction_cost = shape_size_(instruction->shape());
48 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in class:xla::cpu::SimpleCostModel
57 shape_size_(shape_size),
80 instruction_cost = shape_size_(instruction->shape());
103 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in class:xla::cpu::DefaultCostModel

Completed in 56 milliseconds