Searched refs:cond_i (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 598 cond_i = array_ops.gather(cond, i)
599 return array_ops.where(cond_i, a_i, b_i)
618 cond_i = array_ops.gather(cond, i)
619 return array_ops.where_v2(cond_i, a_i, b_i)
638 cond_i = array_ops.gather(cond, i)
639 return array_ops.where_v2(cond_i, a_i, b_i)
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 319 auto cond_i = cond_b.AddInstruction( in GetFinalFftUsingCollectivePermute() local
322 ShapeUtil::MakeShape(PRED, {}), cond_i, in GetFinalFftUsingCollectivePermute()
|
| D | dot_handler.cc | 1408 auto cond_i = cond_b.AddInstruction(HloInstruction::CreateGetTupleElement( in PartitionBaseCase() local
1415 ShapeUtil::MakeShape(PRED, {}), cond_i, in PartitionBaseCase()
|