Home
last modified time | relevance | path

Searched defs:fused (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dzero_batch_test.py45 def testBNWithZeroBatchInputGraph(self, distribution, fused): argument
116 def testBNWithZeroBatchInput(self, distribution, fused): argument
166 def testBNWithDynamicBatchInputEager(self, distribution, fused): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc144 HloInstruction* fused = instr2; in Fuse() local
200 HloInstruction* fused = instr2; in GetNewFusibles() local
247 HloInstruction* fused = instr2; in UpdateBeforeFuse() local
Dlogical_buffer_analysis.cc36 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions() local
Dtuple_points_to_analysis.cc130 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions() local
678 for (auto* fused : instruction->fused_instructions()) { in ToString() local
Dhlo_creation_utils.cc687 HloInstruction* fused, HloInstruction::FusionKind kind) { in MakeFusionInstruction()
Dhlo_verifier.cc2301 for (const auto& fused : fusion->fused_instructions()) { in CheckFusionInstruction() local
Dhlo_instructions.cc1881 for (auto* fused : fused_instructions()) { in IsElementwiseImpl() local
1975 auto* fused = FuseInstruction(instruction); in MergeFusionInstruction() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dfusion.cc134 Value fused = iface.fuse(op.getLoc(), op.set(), rewriter); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.cc139 HloInstruction* fused = candidates[j]; in Run() local
Dmulti_output_fusion.cc240 HloInstruction* fused = *j; in FuseSiblings() local
Djitrt_custom_calls.cc720 Optional<FusedConvAttrs> fused = llvm::None, in GetConvDescriptor()
/external/rust/crates/async-stream/tests/
Dstream.rs52 async fn fused() { in fused() function
/external/rust/crates/peeking_take_while/src/
Dlib.rs210 fn fused() { in fused() function
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/
Dthlo_ops.cc321 Value fused = fuseConcatenateOpThroughPointRecursively( in fuseConcatenateOpThroughPointRecursively() local
333 Value fused = fuseConcatenateOpThroughPointRecursively( in fuseConcatenateOpThroughPointRecursively() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc93 } else if (auto fused = unwrapped_inst_loc.dyn_cast<mlir::FusedLoc>()) { in ConvertLocation() local
/external/tensorflow/tensorflow/core/ir/importexport/
Dconvert_attributes.cc61 } else if (auto fused = inst_loc.dyn_cast<FusedLoc>()) { in ConvertLocation() local
Dgraphdef_export.cc382 if (auto fused = inst_loc.dyn_cast<mlir::FusedLoc>()) { in ExtractExperimentalDebugInfoFromLocation() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dlmhlo_to_jitrt.cc635 if (auto fused = dyn_cast<ConvForwardFusedOp>(op.getOperation())) { in matchAndRewrite() local
641 if (auto fused = dyn_cast<ConvForwardFusedSideInputOp>(op.getOperation())) { in matchAndRewrite() local
/external/skia/src/gpu/ganesh/geometry/
DGrAAConvexTessellator.cpp798 const int fused = fCandidateVerts.fuseWithBoth(); in createInsetRing() local
/external/rust/crates/itertools/src/adaptors/
Dmod.rs522 fused: Option<bool>, field
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc2198 void SetFusedOpAttributes(NodeDef* fused, in SetFusedOpAttributes()
/external/clang/lib/CodeGen/
DCGCall.cpp2595 if (llvm::Value *fused = tryEmitFusedAutoreleaseOfResult(CGF, result)) in emitAutoreleaseOfResult() local