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

  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding.cc 102 auto& operand_indices = pair.second; local
104 if (operand_indices.empty()) {
114 if (std::find(operand_indices.begin(), operand_indices.end(), kLhsIdx) !=
115 operand_indices.end()) {
138 if (std::find(operand_indices.begin(), operand_indices.end(), kRhsIdx) !=
139 operand_indices.end()) {
184 OperandIndices operand_indices =
186 if (!operand_indices.empty())
    [all...]
liveness_util.cc 243 std::vector<int64> operand_indices = user->OperandIndices(operand); local
244 return operand_indices.size() == 1 && operand_indices[0] == 0;
250 const std::vector<int64> operand_indices = user->OperandIndices(operand); local
251 if (operand_indices.size() > 1) {
254 CHECK_EQ(1, operand_indices.size());
255 auto* param = user->to_apply()->parameter_instruction(operand_indices[0]);
347 std::vector<int64> operand_indices = user->OperandIndices(operand); local
348 return operand_indices.size() == 1 && operand_indices[0] == 0
    [all...]
algebraic_simplifier.cc 1098 std::vector<int64> operand_indices = instruction->OperandIndices(operand); local
    [all...]
hlo_evaluator.cc     [all...]
hlo_instruction.cc 3030 std::vector<int64> operand_indices = instruction->OperandIndices(operand); local
    [all...]

Completed in 107 milliseconds