| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stable_sort_expander_test.cc | 109 m::Sort(m::Parameter(0), m::Iota()), 0))); in TEST_F()
155 m::Sort(m::Parameter(0), m::Iota()), 0))); in TEST_F()
187 m::Sort(m::Parameter(0), m::Parameter(1), m::Iota()), 0), in TEST_F()
189 m::Sort(m::Parameter(0), m::Parameter(1), m::Iota()), 1)))); in TEST_F()
253 m::Sort(m::Parameter(0), m::Iota(), m::Iota()), 0))); in TEST_F()
290 m::Sort(m::Parameter(0), m::Iota(), m::Iota()), 0))); in TEST_F()
321 m::Sort(m::Parameter(0), m::Iota()), 0))); in TEST_F()
352 m::Sort(m::Parameter(0), m::Iota()), 0)))); in TEST_F()
|
| D | cholesky_expander.cc | 86 Iota(body_builder, ShapeUtil::MakeShape(S32, row_shape_dims), in CholeskyUnblocked()
89 Iota(body_builder, ShapeUtil::MakeShape(S32, col_shape_dims), in CholeskyUnblocked()
|
| D | dynamic_padder_test.cc | 52 op::Select(op::Lt(op::Iota(), op::Broadcast(op::Parameter())), in ExpectPadded()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | iota_test.cc | 42 Iota(&builder, element_type, num_elements); in TEST_P()
74 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
101 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
126 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
|
| D | constants_test.cc | 186 auto val1 = Iota(&b, F32, 3); in TEST_F()
194 auto tuple = Tuple(&b, {Iota(&b, F32, 3), Iota(&b, F32, 1)}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 43 auto a = Iota(builder, U32, m); in IdentityMatrix()
44 auto b = Iota(builder, U32, n); in IdentityMatrix()
62 auto a = Iota(builder, S32, n); in GetMatrixDiagonal()
63 auto b = Iota(builder, S32, m) + offset; in GetMatrixDiagonal()
101 auto a = Iota(builder, S32, n); in TriangleMask()
102 auto b = Iota(builder, S32, m) + ConstantR0<int32>(builder, diagonal); in TriangleMask()
|
| D | sorting.cc | 32 XlaOp iota_s32 = Iota(builder, iota_shape, last_dim); in TopK()
|
| D | qr.cc | 92 XlaOp iota = Iota(builder, S32, m); in House()
203 auto iota = Reshape(Iota(a.builder(), S32, m), {m, 1}); in QRBlock()
|
| D | quantize.h | 106 XlaOp iota_r1 = Iota(builder, U32, unpack_size);
|
| D | arithmetic.cc | 165 XlaOp iota = Iota(builder, output_type, axis_size); in ArgMinMax()
|
| D | prng.cc | 110 inputs[0] = Iota(builder, U32, size); in GetInputs()
|
| D | slicing.cc | 149 to_concat.push_back(Iota(builder, index_shape, i)); in TorchGather()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_set_diag_op.cc | 65 xla::XlaOp iota_m = xla::Iota(builder, xla::S32, m); in Compile()
66 xla::XlaOp iota_n = xla::Iota(builder, xla::S32, n); in Compile()
|
| D | matrix_band_part_op.cc | 65 xla::XlaOp iota_m = xla::Iota(builder, iota_shape, /*iota_dimension=*/0); in Compile()
66 xla::XlaOp iota_n = xla::Iota(builder, iota_shape, /*iota_dimension=*/1); in Compile()
|
| D | reverse_sequence_op.cc | 76 xla::XlaOp batch_idx = xla::Iota( in Compile()
80 xla::XlaOp forward_idx = xla::Iota( in Compile()
|
| D | transpose_op.cc | 160 auto iota = xla::Iota(ctx->builder(), xla::S32, size); in Compile()
|
| D | extract_image_patches_op.cc | 114 xla::XlaOp iota = xla::Iota(builder, xla::S32, kernel_size * depth); in Compile()
|
| D | diag_op.cc | 44 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal()
|
| D | sequence_ops.cc | 64 xla::Iota(builder, xla::primitive_util::NativeToPrimitiveType<T>(), in CreateRangeTensor()
|
| D | conv_op_helpers.cc | 109 xla::XlaOp input_feature_iota = xla::Iota( in CreateExpandedFilterMask()
111 xla::XlaOp expanded_feature_iota = xla::Iota( in CreateExpandedFilterMask()
|
| D | random_ops.cc | 155 xla::XlaOp indices = xla::Iota(builder, xla::S32, n); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 102 xla::Eq(indices, xla::Iota(builder, iota_shape, axis), broadcast_dims), in OneHot()
|
| /external/apache-xml/src/main/java/org/apache/xml/serializer/ |
| D | HTMLEntities.properties | 144 # Iota 921
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 498 XlaOp Iota(const Shape& shape, int64 iota_dimension);
500 XlaOp Iota(PrimitiveType type, int64 size);
932 friend XlaOp Iota(XlaBuilder* builder, const Shape& shape,
934 friend XlaOp Iota(XlaBuilder* builder, PrimitiveType type, int64 size);
1700 XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64 iota_dimension);
1703 XlaOp Iota(XlaBuilder* builder, PrimitiveType type, int64 size);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 461 LocalOp ComputationBuilder::Iota(PrimitiveType element_type, int64 size) { in Iota() function in xla::swig::ComputationBuilder
462 return xla::Iota(&builder_, element_type, size); in Iota()
467 return xla::Iota(&builder_, shape, dimension); in BroadcastedIota()
|