Home
last modified time | relevance | path

Searched refs:operand_indices (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc121 auto& operand_indices = pair.second; in FoldTransposeIntoConvolution() local
123 if (operand_indices.empty()) { in FoldTransposeIntoConvolution()
133 if (absl::c_linear_search(operand_indices, kLhsIdx)) { in FoldTransposeIntoConvolution()
156 if (absl::c_linear_search(operand_indices, kRhsIdx)) { in FoldTransposeIntoConvolution()
203 OperandIndices operand_indices = in Run() local
205 if (!operand_indices.empty()) { in Run()
206 foldable_dots.emplace_back(instruction, operand_indices); in Run()
210 OperandIndices operand_indices = CanFoldOperandsIntoConvolution( in Run() local
212 if (!operand_indices.empty()) { in Run()
214 std::make_pair(instruction, operand_indices)); in Run()
Dtuple_points_to_analysis.cc766 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
767 return operand_indices.size() == 1 && operand_indices[0] == 0; in CanShareOperandBufferWithUser()
781 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
782 return operand_indices.size() == 1 && user_index[0] == operand_indices[0]; in CanShareOperandBufferWithUser()
789 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
790 return operand_indices.size() == 1 && operand_indices[0] == 1; in CanShareOperandBufferWithUser()
796 const std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
797 if (operand_indices.size() > 1) { in CanShareOperandBufferWithUser()
800 CHECK_EQ(1, operand_indices.size()); in CanShareOperandBufferWithUser()
801 auto* param = user->to_apply()->parameter_instruction(operand_indices[0]); in CanShareOperandBufferWithUser()
Dhlo_dataflow_analysis.cc1099 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
1100 return operand_indices.size() == 1 && operand_indices[0] == 0; in CanShareOperandBufferWithUser()
1114 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
1115 return operand_indices.size() == 1 && user_index[0] == operand_indices[0]; in CanShareOperandBufferWithUser()
Dhlo_instructions.cc45 std::vector<int64> operand_indices = instruction->OperandIndices(operand); in IsInstructionElementwiseOnOperand() local
46 return absl::c_all_of(operand_indices, [instruction](int64 operand_index) { in IsInstructionElementwiseOnOperand()
1580 absl::flat_hash_map<const HloInstruction*, int> operand_indices; in DeduplicateFusionOperands() local
1583 auto emplace_result = operand_indices.emplace(operand(i), i); in DeduplicateFusionOperands()
Dhlo_evaluator_typed_visitor.h2876 std::vector<int64> operand_indices(start.size()); in DynamicSlice()
2880 for (int64 i = 0; i < operand_indices.size(); ++i) { in DynamicSlice()
2882 operand_indices[i] = multi_index[i] + start[i]; in DynamicSlice()
2885 auto result = operand_literal.Get<ReturnT>(operand_indices); in DynamicSlice()
Dalgebraic_simplifier.cc2015 std::vector<int64> operand_indices = instruction->OperandIndices(operand); in OutputIsSubsetOfOperandElements() local
2016 CHECK(!operand_indices.empty()); in OutputIsSubsetOfOperandElements()
2017 if (operand_indices.size() != 1) { in OutputIsSubsetOfOperandElements()
2020 int64 operand_index = operand_indices[0]; in OutputIsSubsetOfOperandElements()
/external/swiftshader/third_party/SPIRV-Tools/source/reduce/
Dstructured_loop_to_selection_reduction_opportunity.cpp141 std::vector<uint32_t> operand_indices; in RedirectEdge() local
143 operand_indices = {0}; in RedirectEdge()
145 operand_indices = {1, 2}; in RedirectEdge()
150 operand_indices.push_back(label_index); in RedirectEdge()
157 for (auto operand_index : operand_indices) { in RedirectEdge()
/external/deqp-deps/SPIRV-Tools/source/reduce/
Dstructured_loop_to_selection_reduction_opportunity.cpp139 std::vector<uint32_t> operand_indices; in RedirectEdge() local
141 operand_indices = {0}; in RedirectEdge()
143 operand_indices = {1, 2}; in RedirectEdge()
148 operand_indices.push_back(label_index); in RedirectEdge()
155 for (auto operand_index : operand_indices) { in RedirectEdge()