Searched refs:operand_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding.cc | 150 auto& operand_indices = pair.second; in FoldTransposeIntoConvolution() local
152 if (operand_indices.empty()) { in FoldTransposeIntoConvolution()
162 if (absl::c_linear_search(operand_indices, kLhsIdx)) { in FoldTransposeIntoConvolution()
185 if (absl::c_linear_search(operand_indices, kRhsIdx)) { in FoldTransposeIntoConvolution()
232 OperandIndices operand_indices = in Run() local
234 if (!operand_indices.empty()) { in Run()
235 foldable_dots.emplace_back(instruction, operand_indices); in Run()
239 OperandIndices operand_indices = CanFoldOperandsIntoConvolution( in Run() local
241 if (!operand_indices.empty()) { in Run()
243 std::make_pair(instruction, operand_indices)); in Run()
|
| D | hlo_dataflow_analysis.cc | 1346 const auto operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
1348 return operand_indices.size() == 1 && operand_indices[0] == operand_no; in CanShareOperandBufferWithUser()
1362 const auto operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
1363 return operand_indices.size() == 1 && user_index[0] == operand_indices[0]; in CanShareOperandBufferWithUser()
|
| D | hlo_instructions.cc | 48 const auto operand_indices = instruction->OperandIndices(operand); in IsInstructionElementwiseOnOperand() local
49 return absl::c_all_of(operand_indices, [instruction](int64 operand_index) { in IsInstructionElementwiseOnOperand()
1846 absl::flat_hash_map<const HloInstruction*, int> operand_indices; in DeduplicateFusionOperands() local
1849 auto emplace_result = operand_indices.emplace(operand(i), i); in DeduplicateFusionOperands()
|
| D | hlo_evaluator_typed_visitor.h | 2903 std::vector<int64> operand_indices(start.size());
2907 for (int64 i = 0; i < operand_indices.size(); ++i) {
2909 operand_indices[i] = multi_index[i] + start[i];
2912 auto result = operand_literal.Get<ReturnT>(operand_indices);
|
| D | algebraic_simplifier.cc | 3026 const auto operand_indices = instruction->OperandIndices(operand); in OutputIsSubsetOfOperandElements() local
3027 CHECK(!operand_indices.empty()); in OutputIsSubsetOfOperandElements()
3028 if (operand_indices.size() != 1) { in OutputIsSubsetOfOperandElements()
3031 int64 operand_index = operand_indices[0]; in OutputIsSubsetOfOperandElements()
|
| /external/deqp-deps/SPIRV-Tools/source/reduce/ |
| D | structured_loop_to_selection_reduction_opportunity.cpp | 134 std::vector<uint32_t> operand_indices; in RedirectEdge() local
136 operand_indices = {0}; in RedirectEdge()
138 operand_indices = {1, 2}; in RedirectEdge()
143 operand_indices.push_back(label_index); in RedirectEdge()
150 for (auto operand_index : operand_indices) { in RedirectEdge()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/reduce/ |
| D | structured_loop_to_selection_reduction_opportunity.cpp | 134 std::vector<uint32_t> operand_indices; in RedirectEdge() local
136 operand_indices = {0}; in RedirectEdge()
138 operand_indices = {1, 2}; in RedirectEdge()
143 operand_indices.push_back(label_index); in RedirectEdge()
150 for (auto operand_index : operand_indices) { in RedirectEdge()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/reduce/ |
| D | structured_loop_to_selection_reduction_opportunity.cpp | 131 std::vector<uint32_t> operand_indices; in RedirectEdge() local
133 operand_indices = {0}; in RedirectEdge()
135 operand_indices = {1, 2}; in RedirectEdge()
140 operand_indices.push_back(label_index); in RedirectEdge()
147 for (auto operand_index : operand_indices) { in RedirectEdge()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1194 for operand_indices, binary_equation in indices_and_equations:
1198 operands = list(map(inputs.pop, operand_indices))
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1302 std::vector<int> operand_indices; in IsStatefulOperand() local
1303 if (!mlir::TFL::IsStatefulOp(op, &operand_indices)) return false; in IsStatefulOperand()
1304 return absl::c_find(operand_indices, operand_index) != operand_indices.end(); in IsStatefulOperand()
|