Home
last modified time | relevance | path

Searched refs:pass_through (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_metadata.cc96 std::vector<PassThrough> pass_through; in LocatePassThroughDomainLinks() local
103 pass_through.emplace_back(user, instruction); in LocatePassThroughDomainLinks()
110 pass_through.emplace_back(nullptr, instruction); in LocatePassThroughDomainLinks()
116 return pass_through; in LocatePassThroughDomainLinks()
121 for (auto& pass_through : LocatePassThroughDomainLinks(domain)) { in FixupPassThroughDomainLinks() local
122 HloInstruction* tuple = pass_through.operand->parent()->AddInstruction( in FixupPassThroughDomainLinks()
123 HloInstruction::CreateTuple({pass_through.operand})); in FixupPassThroughDomainLinks()
124 HloInstruction* gte = pass_through.operand->parent()->AddInstruction( in FixupPassThroughDomainLinks()
125 HloInstruction::CreateGetTupleElement(pass_through.operand->shape(), in FixupPassThroughDomainLinks()
128 if (pass_through.user != nullptr) { in FixupPassThroughDomainLinks()
[all …]
/external/mesa3d/src/mesa/drivers/dri/i915/
Di830_texblend.c44 pass_through(GLuint * state, GLuint blendUnit) in pass_through() function
213 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
242 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
259 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
300 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
329 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
340 return pass_through(state, blendUnit); in i830SetTexEnvCombine()
425 tmp_sz = pass_through(tmp, unit); in emit_passthrough()
/external/llvm-project/mlir/test/Conversion/StandardToLLVM/
Dconvert-funcs.mlir29 //CHECK-LABEL: llvm.func @pass_through(%arg0: !llvm.ptr<func<void ()>>) -> !llvm.ptr<func<void ()>>…
30 func @pass_through(%arg0: () -> ()) -> (() -> ()) {
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py122 pass_through = (larger_shape, larger_shape)
123 shape_pairs = (up_sample, down_sample, pass_through)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py4374 pass_through = i in self._body_pass_through_indices
4380 if not pass_through:
4386 if not pass_through: