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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_parallelization_preparation.h 42 : max_parallelism_(max_parallelism), shape_size_(shape_size) {}
74 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in class:xla::cpu::ParallelizationPreparation
parallel_task_assignment.cc 32 : max_parallelism_(max_parallelism), shape_size_(shape_size) {}
37 const int64 instruction_cost = shape_size_(instruction->shape());
46 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in class:xla::cpu::SimpleCostModel
55 shape_size_(shape_size),
78 instruction_cost = shape_size_(instruction->shape());
101 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in class:xla::cpu::DefaultCostModel
cpu_parallelization_preparation.cc 114 ParallelTaskAssignment parallel_task_assignment(max_parallelism_, shape_size_,
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc 39 : shape_size_(shape_size), per_second_rates_(per_second_rates) {}
51 float bytes_accessed = shape_size_(hlo->shape());
53 bytes_accessed += shape_size_(operand->shape());
185 current_properties_[kBytesAccessedKey] = shape_size_(tuple->shape());
458 current_properties_[kBytesAccessedKey] += shape_size_(subshape);
462 current_properties_[kBytesAccessedKey] += shape_size_(operand->shape());
580 shape_size = &shape_size_;
hlo_cost_analysis.h 149 // given hlo. Uses shape_size_ to calculate shape sizes if shape_size is null,
150 // otherwise uses shape_size_.
173 const ShapeSizeFunction shape_size_; member in class:xla::HloCostAnalysis

Completed in 648 milliseconds