Home
last modified time | relevance | path

Searched refs:scatter_operand_count (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h2167 int64_t scatter_operand_count() const { return operand_count() / 2; } in scatter_operand_count() function
2169 return absl::MakeConstSpan(operands()).first(scatter_operand_count()); in scatter_operands()
2172 return absl::MakeConstSpan(operands()).last(scatter_operand_count()); in scatter_updates()
2175 return operand(scatter_operand_count()); in scatter_indices()
2178 return mutable_operand(scatter_operand_count()); in scatter_indices()
Dscatter_expander.cc254 auto operands = loop_state.first(scatter->scatter_operand_count()); in ScatterLoopBody()
Dhlo_cost_analysis.cc1082 for (int i = 0, n = scatter->scatter_operand_count(); i < n; ++i) { in HandleScatter()
1090 SetOperandBytesAccessed(scatter->scatter_operand_count(), in HandleScatter()
Dhlo_dataflow_analysis.cc1775 if (scatter && scatter->scatter_operand_count() > 1) { in GetInPlaceInputOutputPairs()
1777 pairs.reserve(scatter->scatter_operand_count()); in GetInPlaceInputOutputPairs()
1778 for (int i = 0, n = scatter->scatter_operand_count(); i < n; ++i) { in GetInPlaceInputOutputPairs()
Dhlo_evaluator.cc2844 operands.reserve(scatter->scatter_operand_count()); in HandleScatter()
Dalgebraic_simplifier.cc5973 if (scatter->scatter_operand_count() == 1 && in HandleScatter()