| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | where_op.cc | 64 xla::XlaOp iota = xla::Iota(ctx->builder(), iota_shape, axis); in Compile() local
|
| D | sequence_ops.cc | 162 xla::XlaOp iota = xla::Iota(ctx->builder(), ctx->output_xla_type(0), num); in Compile() local
|
| D | diag_op.cc | 45 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal() local
|
| D | transpose_op.cc | 182 auto iota = in InvertPermutation() local
|
| D | cast_op.cc | 164 auto iota = xla::Iota(ctx->builder(), xla_input_shape, in Compile() local
|
| D | unique_op.cc | 154 xla::XlaOp iota = xla::Iota(ctx->builder(), iota_shape, 0); in DataOutputFastPath() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | topk_rewriter.cc | 75 HloIotaInstruction* iota = in Run() local
|
| D | stable_sort_expander.cc | 75 auto iota = computation->AddInstruction( in ExpandInstruction() local
|
| D | qr_expander.cc | 97 XlaOp iota = Iota(builder, S32, m); in House() local
230 auto iota = Reshape(Iota(a.builder(), S32, m), {m, 1}); in QrBlock() local
|
| D | dynamic_padder.cc | 258 HloInstruction* iota = in PadWithScalar() local
613 HloInstruction* iota = in RewriteDynamicReshapeCombineInput() local
1214 HloInstruction* iota = in RewriteDynamicSort() local
1372 HloInstruction* iota = in RewriteDynamicUpdateSlice() local
|
| D | dfs_hlo_visitor_with_default.h | 149 Status HandleIota(HloInstructionPtr iota) override { in HandleIota()
|
| D | pattern_matcher_test.cc | 680 std::unique_ptr<HloInstruction> iota = in TEST_F() local
|
| D | hlo_sharding_util.cc | 1285 if (auto* iota = DynCast<HloIotaInstruction>(op)) { in GetGatherBatchParallelDims() local
1295 } else if (auto* iota = DynCast<HloIotaInstruction>(indices)) { in GetGatherBatchParallelDims() local
|
| D | algebraic_simplifier_test.cc | 3039 auto iota = builder.AddInstruction(HloInstruction::CreateIota( in TEST_F() local
3060 auto iota = builder.AddInstruction( in TEST_F() local
3083 auto iota = builder.AddInstruction( in TEST_F() local
3108 auto iota = builder.AddInstruction( in TEST_F() local
3129 auto iota = builder.AddInstruction( in TEST_F() local
3149 auto iota = builder.AddInstruction( in TEST_F() local
3171 auto iota = builder.AddInstruction( in TEST_F() local
3194 auto iota = builder.AddInstruction( in TEST_F() local
4721 HloInstruction* iota = in TEST_F() local
4741 HloInstruction* iota = in TEST_F() local
|
| D | algebraic_simplifier.cc | 3311 auto* iota = Cast<HloIotaInstruction>(instruction); in HandleIota() local
3815 HloInstruction* iota; in HandleRemainder() local
3927 auto* iota = Cast<HloIotaInstruction>(operand); in HandleReshape() local
|
| D | hlo_verifier.cc | 606 auto* iota = Cast<HloIotaInstruction>(hlo); in HandleIota() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 159 XlaOp iota = Iota(builder, iota_shape, axis); in ArgMinMax() local
190 XlaOp iota = Iota( in ArgMinMaxTwoPass() local
|
| D | prng.cc | 380 XlaOp iota = Iota(builder, U64, n); in GetPhiloxInputsAndUpdatedState() local
|
| /external/skia/src/opts/ |
| D | SkVM_opts.h | 206 const int iota[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15, in interpret_skvm() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 201 auto iota = add_hlo( in GetCorrectionFactor() local
|
| D | spmd_partitioner_util.cc | 1022 auto iota = b->AddInstruction(HloInstruction::CreateIota(index_shape, dim)); in ExchangeHaloAndGetValidData() local
1164 HloIotaInstruction* iota = in GetKValueInTopKWhenPartitionSortDim() local
|
| D | dot_handler.cc | 3243 auto add_iota = [&](const HloInstruction* iota) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3323 auto iota = body->AddInstruction(HloInstruction::CreateIota( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
|
| D | spmd_partitioner.cc | 493 auto iota = in PadWithValue() local
2151 auto iota = b_.AddInstruction(HloInstruction::CreateIota( in HandleIota() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3439 auto iota = rewriter.create<IotaOp>(op.getLoc(), result_type, in matchAndRewrite() local
3519 auto iota = rewriter.create<DynamicIotaOp>( in matchAndRewrite() local
3590 auto iota = rewriter.create<IotaOp>(op.getLoc(), result_type, in matchAndRewrite() local
4521 auto iota = rewriter.create<IotaOp>( in matchAndRewrite() local
5724 Value iota = builder->create<IotaOp>( in House() local
5870 auto iota = builder->create<IotaOp>( in QRBlock() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1000 int64_t iota = 0; in IsIotaAttr() local
|