Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dzero_batch_test.py48 def testBNWithZeroBatchInputGraph(self, distribution, fused): argument
119 def testBNWithZeroBatchInput(self, distribution, fused): argument
169 def testBNWithDynamicBatchInputEager(self, distribution, fused): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc141 HloInstruction* fused = instr2; in Fuse() local
202 HloInstruction* fused = instr2; in GetNewFusibles() local
249 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
637 for (auto* fused : instruction->fused_instructions()) { in ToString() local
Dtuple_points_to_analysis_test.cc777 fusion->fused_instructions(), [&](const HloInstruction* fused) { in GetFusionParameterForOperand()
Dhlo_verifier.cc1729 for (const auto& fused : fusion->fused_instructions()) { in CheckFusionInstruction() local
Dhlo_instructions.cc1414 for (auto* fused : fused_instructions()) { in IsElementwiseImpl() local
1517 auto* fused = FuseInstruction(instruction); in MergeFusionInstruction() local
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization.py141 fused=None, argument
197 fused=None, argument
Dnormalization_test.py45 def _simple_model(self, image, fused, freeze_mode): argument
/external/rust/crates/async-stream/tests/
Dstream.rs52 async fn fused() { in fused() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcontraction_fusion.cc126 Value fused = rewriter.create<FusedOp>(loc, result_ty, operands, attrs); in matchAndRewrite() local
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_test.py179 def test_batchnorm_mixed_precision_does_not_overflow(self, fused): argument
386 def _run_batchnorm_correctness_test(layer, dtype='float32', fused=False): argument
Dnormalization.py183 fused=None, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.cc129 HloInstruction* fused = candidates[j]; in Run() local
Dmulti_output_fusion.cc229 HloInstruction* fused = *j; in FuseSiblings() local
/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_utils_test.py404 def test_batchnorm_correctness(self, distribution, fused, optimizer): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc80 } else if (auto fused = inst_loc.dyn_cast<mlir::FusedLoc>()) { in ConvertLocation() local
/external/skqp/src/gpu/ops/
DGrAAConvexTessellator.cpp771 const int fused = fCandidateVerts.fuseWithBoth(); in createInsetRing() local
/external/skia/src/gpu/ops/
DGrAAConvexTessellator.cpp787 const int fused = fCandidateVerts.fuseWithBoth(); in createInsetRing() local
/external/rust/crates/itertools/src/adaptors/
Dmod.rs514 fused: Option<bool>, field
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1050 void SetFusedOpAttributes(NodeDef* fused, in IsCpuCompatibleDataType()
/external/llvm-project/polly/lib/External/isl/
Disl_coalesce.c515 struct isl_basic_map *fused = NULL; in fuse() local
/external/clang/lib/CodeGen/
DCGCall.cpp2595 if (llvm::Value *fused = tryEmitFusedAutoreleaseOfResult(CGF, result)) in emitAutoreleaseOfResult() local
/external/llvm-project/clang/lib/CodeGen/
DCGCall.cpp2939 if (llvm::Value *fused = tryEmitFusedAutoreleaseOfResult(CGF, result)) in emitAutoreleaseOfResult() local