Home
last modified time | relevance | path

Searched refs:kMap (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_to_select_test.cc73 ASSERT_EQ(root->opcode(), HloOpcode::kMap); in TEST_F()
126 ASSERT_EQ(root->opcode(), HloOpcode::kMap); in TEST_F()
182 ASSERT_EQ(root->opcode(), HloOpcode::kMap); in TEST_F()
Dhlo_opcode_test.cc63 case HloOpcode::kMap: in TEST()
Dhlo_opcode.h106 V(kMap, "map", kHloOpcodeIsVariadic) \
Dhlo_element_type_converter.cc146 opcode == HloOpcode::kMap || opcode == HloOpcode::kReduce || in Run()
Dhlo_instruction.cc304 case HloOpcode::kMap: in CreateFromProto()
1677 case HloOpcode::kMap: in CloneWithNewOperands()
2214 case HloOpcode::kMap: in IdenticalSlowPath()
2431 case HloOpcode::kMap: in to_apply()
2452 case HloOpcode::kMap: in set_to_apply()
2822 } else if (opcode() == HloOpcode::kCall || opcode() == HloOpcode::kMap || in ExtraAttributesToString()
2881 case HloOpcode::kMap: in ExtraAttributesToString()
3034 case HloOpcode::kMap: in IsFusible()
3148 case HloOpcode::kMap: in Visit()
Dcall_graph.cc62 case HloOpcode::kMap: in GetInstructionCallContext()
Dinstruction_fusion.cc167 case HloOpcode::kMap: in IsExpensive()
Dhlo_module.cc170 case HloOpcode::kMap: in ReplaceComputations()
Dhlo_graph_dumper.cc1060 case HloOpcode::kMap: in GetInstructionColor()
Dbuffer_assignment.cc195 case HloOpcode::kMap: in GatherComputationsByAllocationType()
Dsharding_propagation.cc150 case HloOpcode::kMap: in PickRepresentativeOperand()
Dhlo_verifier.cc44 case HloOpcode::kMap: in IsCallerInstruction()
Dhlo_parser.cc124 case HloOpcode::kMap: in CanInferShape()
1785 case HloOpcode::kMap: { in ParseInstructionRhs()
Dlayout_assignment.cc2237 case HloOpcode::kMap: in InstructionCanChangeLayout()
Dspace_to_batch_converter.cc675 hlo->opcode() == HloOpcode::kIota || hlo->opcode() == HloOpcode::kMap) { in IsTrivialElementwise()
Dhlo_instructions.cc1143 : HloInstruction(HloOpcode::kMap, shape) { in HloMapInstruction()
/external/openscreen/tools/cddl/
Dsema.h29 kMap, enumerator
105 case CddlType::Which::kMap:
246 kMap, enumerator
Dsema.cc50 case CddlType::Which::kMap: in ~CddlType()
179 CddlType* map = AddCddlType(table, CddlType::Which::kMap); in AnalyzeType2()
669 case CddlType::Which::kMap: { in MakeCppType()
672 cpp_type->struct_type.key_type = CppType::Struct::KeyType::kMap; in MakeCppType()
854 case CddlType::Which::kMap: in DumpType()
Dcodegen.cc529 if (cpp_type.struct_type.key_type == CppType::Struct::KeyType::kMap) { in WriteEncoder()
984 if (real_type->struct_type.key_type == CppType::Struct::KeyType::kMap) { in WriteEncoders()
1172 if (cpp_type.struct_type.key_type == CppType::Struct::KeyType::kMap) { in WriteDecoder()
1473 if (real_type->struct_type.key_type == CppType::Struct::KeyType::kMap) { in WriteDecoders()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc521 module.get(), {HloOpcode::kParameter, HloOpcode::kExp, HloOpcode::kMap}); in TEST_F()
546 HloOpcode::kExp, HloOpcode::kExp, HloOpcode::kMap}); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc453 return xla::HloOpcode::kMap; in MhloToHloOpcode()
Dhlo_function_importer.cc682 case HloOpcode::kMap: { in ImportInstructionImpl()
/external/angle/src/libANGLE/renderer/vulkan/
Dvk_format_table_autogen.cpp2367 static constexpr angle::FormatMap<VkFormat> kMap = { in GetVkFormatFromFormatID() local
2517 return kMap[formatID]; in GetVkFormatFromFormatID()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc320 case HloOpcode::kMap: in EmitOp()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc240 if (opcode == HloOpcode::kMap) { in LmhloOpIsElementwise()
5042 case HloOpcode::kMap: in IsInstructionSafeForShmemTranspose()

12