Searched refs:TileOp (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tile_ops.cc | 39 class TileOp : public XlaOpKernel { class
41 explicit TileOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TileOp() function in tensorflow::__anon66d155fc0111::TileOp
106 TF_DISALLOW_COPY_AND_ASSIGN(TileOp);
109 REGISTER_XLA_OP(Name("Tile").CompileTimeConstantInput("multiples"), TileOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_ops.cc | 159 class TileOp : public OpKernel { class
161 explicit TileOp(OpKernelConstruction* context) : OpKernel(context) {} in TileOp() function in tensorflow::TileOp
256 TF_DISALLOW_COPY_AND_ASSIGN(TileOp);
261 inline void TileOp<Device, Tmultiples>::HandleCase( in HandleCase()
274 void TileOp<device, Tmultiples>::HandleCase<dtype>( \
583 TileOp<CPUDevice, int32>);
588 TileOp<CPUDevice, int64>);
607 TileOp<GPUDevice, int32>); \
613 TileOp<GPUDevice, int64>);
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1451 auto tf_tile_op = cast<TF::TileOp>(op); in matchAndRewrite()
1467 rewriter.replaceOpWithNewOp<tosa::TileOp>(op, output_type, tf_tile_op.input(), in matchAndRewrite()
|
| D | legalize_tfl.cc | 1964 auto tfl_tile_op = cast<TFL::TileOp>(op); in matchAndRewrite()
1979 rewriter.replaceOpWithNewOp<tosa::TileOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2257 static LogicalResult Verify(TileOp op) { in Verify()
2311 OpFoldResult TileOp::fold(ArrayRef<Attribute> operands) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3976 class ConvertTileOp : public OpRewritePattern<TF::TileOp> {
3980 LogicalResult matchAndRewrite(TF::TileOp op, in matchAndRewrite()
|