Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc150 auto& operand_indices = pair.second; in FoldTransposeIntoConvolution() local
232 OperandIndices operand_indices = in Run() local
239 OperandIndices operand_indices = CanFoldOperandsIntoConvolution( in Run() local
Dhlo_dataflow_analysis.cc1346 const auto operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
1362 const auto operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser() local
Dhlo_instructions.cc48 const auto operand_indices = instruction->OperandIndices(operand); in IsInstructionElementwiseOnOperand() local
1846 absl::flat_hash_map<const HloInstruction*, int> operand_indices; in DeduplicateFusionOperands() local
Dalgebraic_simplifier.cc3026 const auto operand_indices = instruction->OperandIndices(operand); in OutputIsSubsetOfOperandElements() local
/external/deqp-deps/SPIRV-Tools/source/reduce/
Dstructured_loop_to_selection_reduction_opportunity.cpp134 std::vector<uint32_t> operand_indices; in RedirectEdge() local
/external/swiftshader/third_party/SPIRV-Tools/source/reduce/
Dstructured_loop_to_selection_reduction_opportunity.cpp134 std::vector<uint32_t> operand_indices; in RedirectEdge() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/reduce/
Dstructured_loop_to_selection_reduction_opportunity.cpp131 std::vector<uint32_t> operand_indices; in RedirectEdge() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1302 std::vector<int> operand_indices; in IsStatefulOperand() local