Searched defs:iota (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dextract_image_patches_op.cc111 // iota = np.arange(np.prod(ksize), depth)
112 // filter = np.equal(np.reshape(iota, [-1, 1]), iota).astype(np.float32)
113 xla::ComputationDataHandle iota; variable
115 kernel_size * depth, &iota));
117 auto lhs = builder->Reshape(iota, lhs_shape);
119 builder->Eq(lhs, iota, {num_spatial_dims + 1}), type);
H A Ddiag_op.cc41 xla::ComputationDataHandle iota; local
43 XlaHelpers::Iota(builder, DataType::DT_INT32, last_dim_size, &iota));
45 builder->Broadcast(iota, {last_dim_size});
46 xla::ComputationDataHandle mask = builder->Eq(iota_broadcast, iota, {0});
H A Dreverse_sequence_op.cc158 xla::ComputationDataHandle iota; variable
160 context, XlaHelpers::Iota(builder, seq_lens_type, max_seq_len, &iota));
163 auto mask = builder->Lt(iota, builder->Reshape(seq_lens, dims), {seq_dim_});
/external/skia/src/core/
H A DSkRasterPipeline.cpp168 static const float iota[] = { local
172 this->unchecked_append(SkRasterPipeline::seed_shader, const_cast<float*>(iota));
/external/skqp/src/core/
H A DSkRasterPipeline.cpp168 static const float iota[] = { local
172 this->unchecked_append(SkRasterPipeline::seed_shader, const_cast<float*>(iota));
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_helpers.cc55 std::iota(broadcast_dims.begin(), broadcast_dims.begin() + axis, 0);
56 std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1);
74 xla::ComputationDataHandle iota; local
77 TF_RETURN_IF_ERROR(XlaHelpers::Iota(builder, output_type, axis_size, &iota));
79 builder->And(full_mask, iota, /*broadcast_dimensions=*/{axis});
203 int64 size, xla::ComputationDataHandle* iota) {
222 *iota = builder->ConstantLiteral(linspace_literal);
264 std::iota(broadcast_dims.begin(), broadcast_dims.begin() + axis, 0);
265 std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1);
202 Iota(xla::ComputationBuilder* builder, DataType dtype, int64 size, xla::ComputationDataHandle* iota) argument
/external/skia/src/jumper/
H A DSkJumper_stages_lowp.cpp249 STAGE_GG(seed_shader, const float* iota) { argument
250 x = cast<F>(I32(dx)) + unaligned_load<F>(iota);
/external/skqp/src/jumper/
H A DSkJumper_stages_lowp.cpp249 STAGE_GG(seed_shader, const float* iota) { argument
250 x = cast<F>(I32(dx)) + unaligned_load<F>(iota);
/external/annotation-tools/annotation-file-utilities/
H A Dannotation-file-utilities.jarMETA-INF/ META-INF/MANIFEST.MF annotations/ annotations/el/ annotations/field/ annotations/io/ annotations/io/classfile/ ...

Completed in 371 milliseconds