| /external/cronet/buildtools/third_party/libc++/trunk/include/__numeric/ |
| D | iota.h | 24 iota(_ForwardIterator __first, _ForwardIterator __last, _Tp __value) in iota() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 46 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal() local
|
| D | transpose_op.cc | 183 auto iota = in InvertPermutation() local
|
| D | sequence_ops.cc | 176 xla::XlaOp iota = xla::Iota(ctx->builder(), ctx->output_xla_type(0), num); in Compile() local
|
| D | approx_topk_op.cc | 84 xla::XlaOp iota = xla::Iota( in Compile() local
|
| D | where_op.cc | 169 XlaOp iota = xla::Iota(ctx->builder(), iota_shape, axis); in CompileWhereWithSort() local
|
| D | unique_op.cc | 174 auto iota = xla::Iota(ctx->builder(), xla::S32, leading_size); in CompileWithAxis() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bitcast_dtypes_expander.cc | 83 XlaOp iota = Iota(&b, input_shape, input_shape.dimensions_size() - 1); in ExpandInstruction() local
|
| D | stable_sort_expander.cc | 75 auto iota = computation->AddInstruction( in ExpandInstruction() local
|
| D | qr_expander.cc | 125 XlaOp iota = Iota(builder, S32, m); in House() local
254 auto iota = Reshape(Iota(a.builder(), S32, m), {m, 1}); in QrBlock() local
|
| D | topk_rewriter.cc | 126 HloIotaInstruction* iota = in SortIsInTopK() local
|
| D | dynamic_padder.cc | 298 HloInstruction* iota = in PadWithScalar() local
334 HloInstruction* iota = in GenerateBinaryMask() local
678 HloInstruction* iota = in RewriteDynamicReshapeCombineInput() local
1363 HloInstruction* iota = in RewriteDynamicSort() local
1630 HloInstruction* iota = in RewriteDynamicUpdateSlice() local
|
| D | dfs_hlo_visitor_with_default.h | 162 Status HandleIota(HloInstructionPtr iota) override { in HandleIota()
|
| D | pattern_matcher_test.cc | 738 std::unique_ptr<HloInstruction> iota = in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 3447 auto iota = builder.AddInstruction(HloInstruction::CreateIota( in TEST_F() local
3468 auto iota = builder.AddInstruction( in TEST_F() local
3491 auto iota = builder.AddInstruction( in TEST_F() local
3516 auto iota = builder.AddInstruction( in TEST_F() local
3537 auto iota = builder.AddInstruction( in TEST_F() local
3557 auto iota = builder.AddInstruction( in TEST_F() local
3579 auto iota = builder.AddInstruction( in TEST_F() local
3602 auto iota = builder.AddInstruction( in TEST_F() local
5134 HloInstruction* iota = in TEST_F() local
5154 HloInstruction* iota = in TEST_F() local
|
| D | algebraic_simplifier.cc | 1622 HloInstruction* iota = constant->AddInstruction( in HandleConstant() local
3694 auto* iota = Cast<HloIotaInstruction>(instruction); in HandleIota() local
4191 HloInstruction* iota; in HandleRemainder() local
4303 auto* iota = Cast<HloIotaInstruction>(operand); in HandleReshape() local
|
| /external/cronet/buildtools/third_party/libc++/trunk/test/std/ranges/range.adaptors/range.take/ |
| D | adaptor.pass.cpp | 174 auto iota = std::views::iota(1, 8); in test() local
|
| /external/skia/src/opts/ |
| D | SkVM_opts.h | 271 const int iota[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15, in interpret_skvm() local
|
| /external/cronet/buildtools/third_party/libc++/trunk/test/std/ranges/range.adaptors/range.drop/ |
| D | adaptor.pass.cpp | 171 auto iota = std::views::iota(1, 8); in test() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1280 bool MatchSingleIota(DenseIntElementsAttr dimensions, Value iota) { in MatchSingleIota()
1295 bool MatchIotaBroadCastInDim(DenseIntElementsAttr dimensions, Value iota) { in MatchIotaBroadCastInDim()
1314 bool MatchReshapedIota(DenseIntElementsAttr dimensions, Value iota) { in MatchReshapedIota()
1348 bool MatchConstIotaBroadCastInDim(DenseIntElementsAttr dimensions, Value iota) { in MatchConstIotaBroadCastInDim()
1501 bool MatchIotaConst(DenseIntElementsAttr dimensions, Value iota) { in MatchIotaConst()
1532 bool MatchIota(DenseIntElementsAttr dimensions, Value iota) { in MatchIota()
2062 Value iota = reduce_op.operands().back(); in matchAndRewrite() local
2751 int64_t iota = 0; in IsIotaAttr() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 153 XlaOp iota = Iota(builder, iota_shape, axis); in ArgMinMax() local
|
| D | prng.cc | 381 XlaOp iota = Iota(builder, U64, n); in GetPhiloxInputsAndUpdatedState() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 348 HloInstruction* iota = b_.AddInstruction(HloInstruction::CreateIota( in HandleCustomCallSPMDInternal_RotateRight() local
|
| D | fft_handler.cc | 199 auto iota = add_hlo( in GetCorrectionFactor() local
|
| D | spmd_partitioner_util.cc | 984 auto iota = b->AddInstruction(HloInstruction::CreateIota(index_shape, dim)); in ExchangeHaloAndGetValidData() local
1126 HloIotaInstruction* iota = in GetKValueInTopKWhenPartitionSortDim() local
|