Home
last modified time | relevance | path

Searched refs:kDot (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_expensive_invariant_code_motion_test.cc73 return instr.opcode() == HloOpcode::kDot; in TEST_F()
94 return instr.opcode() == HloOpcode::kDot || in TEST_F()
163 return instr.opcode() == HloOpcode::kDot; in TEST_F()
178 return instr.opcode() == HloOpcode::kDot || in TEST_F()
229 return instr.opcode() == HloOpcode::kDot; in TEST_F()
Dhlo_graph_dumper_test.cc85 RenderedGraphFormat::kDot)); in TEST_F()
109 RenderedGraphFormat::kDot)); in TEST_F()
123 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
142 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
161 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
Dhlo_graph_dumper.h50 kDot, enumerator
Dconvert_operand_folding.cc36 if (instruction->opcode() != HloOpcode::kDot && in InstructionMatchesPattern()
Doperand_upcaster.cc26 case HloOpcode::kDot: in MaybeInferShape()
Dhlo_matchers.h157 : HloMatcher(HloOpcode::kDot, /*operands=*/{lhs, rhs}), in HloDotWithContractingDimsMatcher()
422 new ::xla::testing::HloMatcher(::xla::HloOpcode::kDot, {})); in Dot()
429 ::xla::HloOpcode::kDot, {lhs_matcher, rhs_matcher})); in Dot()
Dbatch_dot_simplification.cc118 return instr->opcode() == HloOpcode::kDot; in Run()
Dfusion_node_indexing_evaluation.cc52 case HloOpcode::kDot: in OpInvalidatesCache()
Dhlo_opcode.h83 V(kDot, "dot", 2) \
Dhlo_cse.cc129 case HloOpcode::kDot: in CseHash()
Ddot_decomposer.cc177 if (instruction->opcode() != HloOpcode::kDot) { in Run()
Dtranspose_folding.cc38 if (HloOpcode::kDot != dot.opcode()) { in CanFoldOperandsIntoDot()
Dbfloat16_normalization_test.cc47 if (hlo.opcode() == HloOpcode::kDot) { in SupportsBF16Operand()
57 hlo.opcode() == HloOpcode::kDot || hlo.opcode() == HloOpcode::kTuple || in SupportsBF16Output()
Dsharding_propagation.cc248 case HloOpcode::kDot: in PickRepresentativeOperand()
304 case HloOpcode::kDot: in SupportSpatialPartitioning()
856 case HloOpcode::kDot: { in InferShardingFromOperands()
1303 case HloOpcode::kDot: { in GetShardingFromUser()
Dhlo_graph_dumper.cc1042 case HloOpcode::kDot: in GetInstructionColor()
1698 case RenderedGraphFormat::kDot: in WrapDotInFormat()
1728 /*debug_options=*/{}, xla::RenderedGraphFormat::kDot, in RegisterFusionState()
Dhlo_replication_analysis.cc142 hlo->opcode() == HloOpcode::kDot || // in DetermineHloInstructionIsReplicated()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc48 hlo->opcode() == HloOpcode::kDot && hlo_shape.dimensions_size() <= 1 && in IsNonComplexNonBatchedMatrixVectorDot()
141 if (consumer->opcode() == HloOpcode::kDot) { in ShouldFuse()
Ddot_op_emitter_internal.h44 CHECK_EQ(instr.opcode(), HloOpcode::kDot); in DotInfo()
Dcpu_layout_assignment.cc102 } else if (instr.opcode() == HloOpcode::kDot) { in OperandsAndResultMustHaveRowMajorLayout()
Ddot_op_emitter.cc77 CHECK_EQ(instr.opcode(), HloOpcode::kDot); in DotInfo()
884 if (hlo.opcode() == HloOpcode::kDot && hlo.shape().dimensions_size() <= 1) { in ProfitableToMakeDotOperandColumnMajor()
911 if (fusion_root_op->opcode() != HloOpcode::kDot) { in ProfitableToMakeDotOperandColumnMajor()
/external/protobuf/src/google/protobuf/io/
Dio_win32_unittest.cc224 static const wstring kDot(L"."); in DeleteAllUnder() local
247 if (kDot != childname && kDotDot != childname) { in DeleteAllUnder()
/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.cc160 /*debug_options=*/{}, xla::RenderedGraphFormat::kDot, in GetCompilerIr()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc111 RenderedGraphFormat::kDot); in GetComputationHloDotGraph()
359 RenderedGraphFormat::kDot); in BuildXlaCompilerSubmodule()
/external/libtextclassifier/native/utils/tflite/
Dstring_projection.cc49 constexpr char kDot = '.'; variable
336 if (c == kComma || c == kDot) { in FindNextSeparator()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc445 return xla::HloOpcode::kDot; in MhloToHloOpcode()
484 return xla::HloOpcode::kDot; in MhloToHloOpcode()

12