Home
last modified time | relevance | path

Searched refs:iota (Results 1 – 25 of 143) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dqr.cc82 std::iota(batch_dim_ids.begin(), batch_dim_ids.end(), 0); in House()
92 XlaOp iota = Iota(builder, S32, m); in House() local
93 XlaOp x_after_k = Mul(x, ConvertElementType(Gt(iota, k), type), in House()
110 auto e_k = Broadcast(ConvertElementType(Eq(iota, k), type), in House()
174 std::iota(batch_dim_indices.begin(), batch_dim_indices.end(), 0); in QRBlock()
203 auto iota = Reshape(Iota(a.builder(), S32, m), {m, 1}); in QRBlock() local
204 auto predecessor_mask = ConvertElementType(Lt(iota, j), type); in QRBlock()
205 auto mask = Broadcast(ConvertElementType(Eq(iota, j), type), in QRBlock()
256 std::iota(batch_dim_indices.begin(), batch_dim_indices.end(), 0); in ComputeWYRepresentation()
Darithmetic.cc121 std::iota(all_dimensions.begin(), all_dimensions.end(), 0); in Any()
145 std::iota(broadcast_dims.begin(), broadcast_dims.begin() + axis, 0); in ArgMinMax()
146 std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1); in ArgMinMax()
165 XlaOp iota = Iota(builder, output_type, axis_size); in ArgMinMax() local
166 XlaOp product = And(full_mask, iota, /*broadcast_dimensions=*/{axis}); in ArgMinMax()
Dquantize.h125 std::iota(shift_transpose_dimensions.begin(),
160 std::iota(transpose_dimensions.begin(), transpose_dimensions.end(), 1);
177 std::iota(result_dimensions.begin(), result_dimensions.end(), 0);
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc384 std::iota(input_data.begin(), input_data.end(), 0.0); in XLA_TEST_F()
404 std::iota(input_data.begin(), input_data.end(), 1.0); in XLA_TEST_F()
409 std::iota(filter_data.begin(), filter_data.end(), 1.0); in XLA_TEST_F()
438 std::iota(filter_data.begin(), filter_data.end(), 1.0); in XLA_TEST_F()
469 std::iota(filter_data.begin(), filter_data.end(), 1.0); in XLA_TEST_F()
498 std::iota(filter_data.begin(), filter_data.end(), 1.0); in XLA_TEST_F()
516 std::iota(input_data.begin(), input_data.end(), 0.0); in XLA_TEST_F()
542 std::iota(input_data.begin(), input_data.end(), 0.0); in XLA_TEST_F()
568 std::iota(input_data.begin(), input_data.end(), 0.0); in XLA_TEST_F()
634 std::iota(input_data.begin(), input_data.end(), 0.0); in XLA_TEST_F()
[all …]
Dround_trip_transfer_test.cc65 std::iota(values.begin(), values.end(), 1.0); in TEST_F()
71 std::iota(values.begin(), values.end(), 1.0); in TEST_F()
77 std::iota(values.begin(), values.end(), 1.0); in TEST_F()
83 std::iota(values.begin(), values.end(), 1.0); in TEST_F()
/external/libbrillo/brillo/
Dsecure_blob_unittest.cc102 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
125 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
141 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
185 std::iota(blob1.begin(), blob1.end(), 0); in TEST_F()
186 std::iota(blob2.begin(), blob2.end(), 32); in TEST_F()
Ddata_encoding_unittest.cc57 std::iota(data3.begin(), data3.end(), 0); // Fills the buffer with 0x00-0xFF. in TEST()
85 std::iota(data3.begin(), data3.end(), 0); // Fills the buffer with 0x00-0xFF. in TEST()
120 std::iota(decoded3.begin(), decoded3.end(), 0); // Fill with 0x00..0xFF. in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc77 std::iota(ratio_broadcast_indices.begin(), ratio_broadcast_indices.end(), 0); in BilinearWeights()
265 std::iota(reshape_dims.begin(), reshape_dims.end(), 0); in CalculateGradData()
273 std::iota(reshaped_weights_indices.begin(), reshaped_weights_indices.end(), in CalculateGradData()
283 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in CalculateGradData()
293 std::iota(grad_output_indices.begin(), grad_output_indices.end(), 0); in CalculateGradData()
459 std::iota(reshaped_dims.begin(), reshaped_dims.end(), 0); in CalculateGradWarp()
584 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in Compile()
669 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in Compile()
Dtranspose_op.cc160 auto iota = xla::Iota(ctx->builder(), xla::S32, size); in Compile() local
161 auto result = XlaScatter(iota, iota, indices, in Compile()
Dextract_image_patches_op.cc114 xla::XlaOp iota = xla::Iota(builder, xla::S32, kernel_size * depth); in Compile() local
116 auto lhs = xla::Reshape(iota, lhs_shape); in Compile()
118 xla::Eq(lhs, iota, {num_spatial_dims + 1}), type); in Compile()
Ddiag_op.cc44 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal() local
45 xla::XlaOp iota_broadcast = xla::Broadcast(iota, {last_dim_size}); in CreateDiagonal()
46 xla::XlaOp mask = xla::Eq(iota_broadcast, iota, {0}); in CreateDiagonal()
Dbias_ops.cc107 std::iota(reduce_dims.begin(), reduce_dims.begin() + feature_dim, 0); in Compile()
108 std::iota(reduce_dims.begin() + feature_dim, reduce_dims.end(), in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc684 std::unique_ptr<HloInstruction> iota = in TEST() local
693 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithName("foo"), in TEST()
697 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithOpcode(HloOpcode::kAdd), in TEST()
706 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithNumOperands(42), in TEST()
710 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithShape(m::Shape().IsTuple()), in TEST()
718 iota, m::Op().WithOperand(2, m::Op().WithOpcode(HloOpcode::kAdd)), in TEST()
736 iota, m::Op().WithFusionKind(HloInstruction::FusionKind::kLoop), in TEST()
741 iota, m::Op().WithTupleIndex(42), in TEST()
745 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().IsConstantScalar(), in TEST()
769 constant, m::Op().Is(iota.get()), in TEST()
[all …]
Ddynamic_padder.cc157 HloInstruction* iota = computation->AddInstruction( in Run() local
165 pred_shape, iota, broadcasted_effective_size, in Run()
/external/libcxx/test/std/numerics/numeric.ops/inclusive.scan/
Dinclusive_scan_op_init.pass.cpp78 std::iota(v.begin(), v.end(), 0); in basic_tests()
86 std::iota(v.begin(), v.end(), 1); in basic_tests()
101 std::iota(v.begin(), v.end(), static_cast<unsigned char>(1)); in basic_tests()
Dinclusive_scan.pass.cpp73 std::iota(v.begin(), v.end(), 0); in basic_tests()
81 std::iota(v.begin(), v.end(), 1); in basic_tests()
Dinclusive_scan_op.pass.cpp79 std::iota(v.begin(), v.end(), 0); in basic_tests()
87 std::iota(v.begin(), v.end(), 1); in basic_tests()
/external/libcxx/test/std/numerics/numeric.ops/transform.inclusive.scan/
Dtransform_inclusive_scan_bop_uop_init.pass.cpp105 std::iota(v.begin(), v.end(), 0); in basic_tests()
113 std::iota(v.begin(), v.end(), 1); in basic_tests()
128 std::iota(v.begin(), v.end(), static_cast<unsigned char>(1)); in basic_tests()
Dtransform_inclusive_scan_bop_uop.pass.cpp96 std::iota(v.begin(), v.end(), 0); in basic_tests()
104 std::iota(v.begin(), v.end(), 1); in basic_tests()
/external/libcxx/test/std/numerics/numeric.ops/transform.exclusive.scan/
Dtransform_exclusive_scan_init_bop_uop.pass.cpp105 std::iota(v.begin(), v.end(), 0); in basic_tests()
113 std::iota(v.begin(), v.end(), 1); in basic_tests()
128 std::iota(v.begin(), v.end(), static_cast<unsigned char>(1)); in basic_tests()
/external/libcxx/test/std/numerics/numeric.ops/exclusive.scan/
Dexclusive_scan.pass.cpp73 std::iota(v.begin(), v.end(), 0); in basic_tests()
81 std::iota(v.begin(), v.end(), 1); in basic_tests()
/external/puffin/src/
Dstream_unittest.cc117 std::iota(buf1.begin(), buf1.end(), 0); in TestWrite()
161 std::iota(buf.begin(), buf.end(), 0); in TEST_F()
186 std::iota(buf.begin(), buf.end(), 0); in TEST_F()
251 std::iota(buf.begin(), buf.end(), 0); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc91 std::iota(broadcast_dims.begin(), broadcast_dims.begin() + axis, 0); in OneHot()
92 std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1); in OneHot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc83 std::iota(dimension_order.rbegin(), dimension_order.rend(), 0); in RowMajorShape()
91 std::iota(dimension_order.begin(), dimension_order.end(), 0); in ColMajorShape()
/external/libcxx/test/std/numerics/numeric.ops/numeric.iota/
Diota.pass.cpp27 std::iota(InIter(ia), InIter(ia+s), 5); in test()

123456