Home
last modified time | relevance | path

Searched refs:kSlice (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dslice_sinker.cc35 CHECK_EQ(slice_1->opcode(), HloOpcode::kSlice); in SameSliceConfiguration()
36 CHECK_EQ(slice_2->opcode(), HloOpcode::kSlice); in SameSliceConfiguration()
51 return operand->opcode() != HloOpcode::kSlice; in IsElementwiseOperationOnSimilarSlices()
97 if (operand_slice->opcode() != HloOpcode::kSlice || in IsSimilarOperationOnSlices()
149 if (operand_slice0->opcode() != HloOpcode::kSlice) { in FindElementwiseOperationGroup()
Dfusion_node_indexing_evaluation_test.cc166 EXPECT_EQ(slice1_0->opcode(), HloOpcode::kSlice); in TEST_F()
173 EXPECT_EQ(slice1_1->opcode(), HloOpcode::kSlice); in TEST_F()
Dbfloat16_support.cc96 case HloOpcode::kSlice: in EffectiveOperandPrecisionIsOutputPrecision()
Dhlo_opcode.h137 V(kSlice, "slice", 1) \
Dwhile_loop_invariant_code_motion.cc118 case HloOpcode::kSlice: in NotWorthHoistingIndividually()
Dhlo_replication_analysis.cc113 hlo->opcode() == HloOpcode::kSlice || // in DetermineHloInstructionIsReplicated()
Dconditional_simplifier.cc104 case HloOpcode::kSlice: in TryRemoveConditional()
Dinstruction_fusion.cc111 case HloOpcode::kSlice: in IsExpensive()
Dhlo_instruction.cc289 case HloOpcode::kSlice: { in CreateFromProto()
1478 case HloOpcode::kSlice: in CloneWithNewOperands()
1947 case HloOpcode::kSlice: in IdenticalSlowPath()
2871 case HloOpcode::kSlice: in Visit()
3237 case HloOpcode::kSlice: in OperandElementUse()
Dalgebraic_simplifier.cc835 if (operands[i]->opcode() != HloOpcode::kSlice || in HandleConcatenate()
842 while (j < operands.size() && operands[j]->opcode() == HloOpcode::kSlice && in HandleConcatenate()
2435 case HloOpcode::kSlice: in OutputIsSubsetOfOperandElements()
3298 CHECK_EQ(slice->opcode(), HloOpcode::kSlice); in TryToReorderSliceAndReshape()
3375 if (slice->operand(0)->opcode() == HloOpcode::kSlice && in HandleSlice()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_trivial_slice.cc33 CHECK(op.type == OperatorType::kSlice); in IsSliceTrivial()
57 if (slice_op->type != OperatorType::kSlice) { in Run()
Dresolve_slice_attributes.cc33 if (slice_op->type != OperatorType::kSlice) return ::tensorflow::Status::OK(); in Run()
Dresolve_constant_slice.cc95 if (base_op->type != OperatorType::kSlice) { in Run()
Dhardcode_min_max.cc466 case OperatorType::kSlice: in Run()
Dquantize.cc94 OperatorType::kSlice, in SupportsQuantization()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc165 {{OperatorType::kSlice, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
166 {{OperatorType::kSlice, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
167 {{OperatorType::kSlice, 3}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
/external/webrtc/webrtc/modules/rtp_rtcp/source/
Drtp_format_h264_unittest.cc26 kSlice = 1, enumerator
472 0, 0x02, kSlice, 0xFF, 0, 0x03, kSlice, 0xFF, in TEST_F()
473 0x00, 0, 0x04, kSlice, 0xFF, 0x00, 0x11}; in TEST_F()
483 EXPECT_EQ(kSlice, payload.type.Video.codecHeader.H264.nalu_type); in TEST_F()
Drtp_format_h264.cc23 kSlice = 1, enumerator
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc41 hlo.opcode() == HloOpcode::kSlice || in CanBeLoopFused()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc249 if (root->opcode() == HloOpcode::kSlice) { in IsInputFusibleSlices()
258 return instr->opcode() == HloOpcode::kSlice && in IsInputFusibleSlices()
Dfusion_merger.cc72 return instruction->opcode() == HloOpcode::kSlice || in CalculateBytesReadByFusionParameter()
Dgpu_fusible.cc209 instr.opcode() == HloOpcode::kSlice || in IsLoopFusible()
Dir_emitter_unnested.cc323 if (root.opcode() == HloOpcode::kSlice) { in GetConsistentInputShapeForRootSlices()
454 case HloOpcode::kSlice: { in HandleFusion()
3155 if (slice_or_tuple->opcode() == HloOpcode::kSlice) { in EmitElementForInputFusibleSlices()
3203 ShapeIndex shape_index = (slice_or_tuple->opcode() == HloOpcode::kSlice) in EmitElementForInputFusibleSlices()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h98 kSlice, enumerator
1486 SliceOperator() : Operator(OperatorType::kSlice) {}
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc354 case HloOpcode::kSlice: { in ImportInstruction()

12