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

  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding.cc 121 auto& operand_indices = pair.second; local
123 if (operand_indices.empty()) {
133 if (absl::c_linear_search(operand_indices, kLhsIdx)) {
156 if (absl::c_linear_search(operand_indices, kRhsIdx)) {
203 OperandIndices operand_indices =
205 if (!operand_indices.empty()) {
206 foldable_dots.emplace_back(instruction, operand_indices);
210 OperandIndices operand_indices = CanFoldOperandsIntoConvolution(
212 if (!operand_indices.empty()) {
214 std::make_pair(instruction, operand_indices));
    [all...]
tuple_points_to_analysis.cc 766 std::vector<int64> operand_indices = user->OperandIndices(operand); local
767 return operand_indices.size() == 1 && operand_indices[0] == 0;
781 std::vector<int64> operand_indices = user->OperandIndices(operand); local
782 return operand_indices.size() == 1 && user_index[0] == operand_indices[0];
789 std::vector<int64> operand_indices = user->OperandIndices(operand); local
790 return operand_indices.size() == 1 && operand_indices[0] == 1;
796 const std::vector<int64> operand_indices = user->OperandIndices(operand) local
    [all...]
hlo_dataflow_analysis.cc 1099 std::vector<int64> operand_indices = user->OperandIndices(operand); local
1114 std::vector<int64> operand_indices = user->OperandIndices(operand); local
    [all...]
algebraic_simplifier.cc     [all...]
hlo_evaluator_typed_visitor.h     [all...]
hlo_instructions.cc 45 std::vector<int64> operand_indices = instruction->OperandIndices(operand); local
46 return absl::c_all_of(operand_indices, [instruction](int64 operand_index) {
    [all...]
  /external/swiftshader/third_party/SPIRV-Tools/source/reduce/
structured_loop_to_selection_reduction_opportunity.cpp 141 std::vector<uint32_t> operand_indices; local
143 operand_indices = {0};
145 operand_indices = {1, 2};
150 operand_indices.push_back(label_index);
157 for (auto operand_index : operand_indices) {
  /external/deqp-deps/SPIRV-Tools/source/reduce/
structured_loop_to_selection_reduction_opportunity.cpp 139 std::vector<uint32_t> operand_indices; local
141 operand_indices = {0};
143 operand_indices = {1, 2};
148 operand_indices.push_back(label_index);
155 for (auto operand_index : operand_indices) {

Completed in 1916 milliseconds