Home
last modified time | relevance | path

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

12345678910>>...13

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlegalize-to-std.mlir114 // CHECK-LABEL: func @iota.const.1() -> tensor<4xi32> {
115 func @iota.const.1() -> tensor<4xi32> {
117 %0 = "mhlo.iota"() {iota_dimension = 0 : i64} : () -> tensor<4xi32>
122 // CHECK-LABEL: func @iota.const.2() -> tensor<2x4xi32> {
123 func @iota.const.2() -> tensor<2x4xi32> {
125 %0 = "mhlo.iota"() {iota_dimension = 0 : i64} : () -> tensor<2x4xi32>
130 // CHECK-LABEL: func @iota.const.3() -> tensor<2x4xi32> {
131 func @iota.const.3() -> tensor<2x4xi32> {
133 %0 = "mhlo.iota"() {iota_dimension = 1 : i64} : () -> tensor<2x4xi32>
138 // CHECK-LABEL: func @iota.const.4() -> tensor<2x3x4xi32> {
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc108 std::iota(all_dimensions.begin(), all_dimensions.end(), 0); in Any()
159 XlaOp iota = Iota(builder, iota_shape, axis); in ArgMinMax() local
164 XlaOp max_argmax = Reduce(builder, {input, iota}, in ArgMinMax()
190 XlaOp iota = Iota( in ArgMinMaxTwoPass() local
195 std::iota(broadcast_dims.begin(), broadcast_dims.begin() + axis, 0); in ArgMinMaxTwoPass()
196 std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1); in ArgMinMaxTwoPass()
200 /*on_true=*/iota, in ArgMinMaxTwoPass()
209 /*on_true=*/iota, in ArgMinMaxTwoPass()
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/
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()
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 …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc75 HloIotaInstruction* iota = in Run() local
82 if (iota == nullptr || iota->shape().rank() != data->shape().rank() || in Run()
83 iota->shape().element_type() != S32 || in Run()
84 iota->opcode() != HloOpcode::kIota || in Run()
85 iota->iota_dimension() != sort->sort_dimension()) { in Run()
Dqr_expander.cc88 std::iota(batch_dim_ids.begin(), batch_dim_ids.end(), 0); in House()
97 XlaOp iota = Iota(builder, S32, m); in House() local
98 XlaOp x_after_k = Mul(x, ConvertElementType(Gt(iota, k), type), in House()
142 auto e_k = Broadcast(ConvertElementType(Eq(iota, k), type), in House()
196 std::iota(batch_dim_indices.begin(), batch_dim_indices.end(), 0); in QrBlock()
230 auto iota = Reshape(Iota(a.builder(), S32, m), {m, 1}); in QrBlock() local
231 auto predecessor_mask = ConvertElementType(Lt(iota, j), type); in QrBlock()
232 auto mask = Broadcast(ConvertElementType(Eq(iota, j), type), in QrBlock()
244 std::iota(dim_ids.begin(), dim_ids.end(), 0); in QrBlock()
251 std::iota(tau_broadcast_dims.begin(), tau_broadcast_dims.end(), 0); in QrBlock()
[all …]
Dpattern_matcher_test.cc680 std::unique_ptr<HloInstruction> iota = in TEST_F() local
689 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithName("foo"), in TEST_F()
693 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithOpcode(HloOpcode::kAdd), in TEST_F()
702 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithNumOperands(42), in TEST_F()
706 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().WithShape(m::Shape().IsTuple()), in TEST_F()
714 iota, m::Op().WithOperand(2, m::Op().WithOpcode(HloOpcode::kAdd)), in TEST_F()
732 iota, m::Op().WithFusionKind(HloInstruction::FusionKind::kLoop), in TEST_F()
737 iota, m::Op().WithTupleIndex(42), in TEST_F()
741 EXPECT_DESC_AND_EXPLANATION(iota, m::Op().IsConstantScalar(), in TEST_F()
765 constant, m::Op().Is(iota.get()), in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dfingerprint_op_test.cc77 std::iota(buffer.data(), buffer.data() + buffer.size(), in TEST_F()
95 std::iota(&buffer(0)[0], &buffer(0)[0] + buffer(0).size(), 0); in TEST_F()
96 std::iota(&buffer(1)[0], &buffer(1)[0] + buffer(1).size(), 7); in TEST_F()
97 std::iota(&buffer(2)[0], &buffer(2)[0] + buffer(2).size(), 71); in TEST_F()
98 std::iota(&buffer(3)[0], &buffer(3)[0] + buffer(3).size(), 41); in TEST_F()
/external/libbrillo/brillo/
Ddata_encoding_test.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()
Dsecure_blob_test.cc103 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
126 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
142 std::iota(std::begin(from_data), std::end(from_data), 0); in TEST_F()
186 std::iota(blob1.begin(), blob1.end(), 0); in TEST_F()
187 std::iota(blob2.begin(), blob2.end(), 32); in TEST_F()
/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()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtranspose_op.cc182 auto iota = in InvertPermutation() local
185 auto result = XlaScatter(iota, iota, indices, in InvertPermutation()
Ddiag_op.cc45 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal() local
46 xla::XlaOp iota_broadcast = xla::Broadcast(iota, {last_dim_size}); in CreateDiagonal()
47 xla::XlaOp mask = xla::Eq(iota_broadcast, iota, {0}); in CreateDiagonal()
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()
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()
Dwhere_op.cc64 xla::XlaOp iota = xla::Iota(ctx->builder(), iota_shape, axis); in Compile() local
65 xla::XlaOp reshaped = xla::Reshape(iota, {flattened_size}); in Compile()
/external/llvm-project/libcxx/test/std/numerics/numeric.ops/inclusive.scan/
Dinclusive_scan_op_init.pass.cpp99 std::iota(v.begin(), v.end(), 0); in basic_tests()
107 std::iota(v.begin(), v.end(), 1); in basic_tests()
132 std::iota(v.begin(), v.end(), static_cast<unsigned char>(1)); in basic_tests()
/external/llvm-project/pstl/test/std/algorithms/alg.modifying.operations/
Dswap_ranges.pass.cpp83 iota(data_b, data_e, 0); in operator ()()
84 iota(actual_b, actual_e, std::distance(data_b, data_e)); in operator ()()
/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/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()
/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/llvm-project/libcxx/test/std/numerics/numeric.ops/exclusive.scan/
Dexclusive_scan.pass.cpp93 std::iota(v.begin(), v.end(), 0); in basic_tests()
101 std::iota(v.begin(), v.end(), 1); in basic_tests()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_key_value_sort.cc54 std::iota(indices.get(), indices.get() + sort_dimension_elements, 0); in __xla_cpu_runtime_KeyValueSort()
61 std::iota(indices.get(), indices.get() + sort_dimension_elements, 0); in __xla_cpu_runtime_KeyValueSort()

12345678910>>...13