| /external/tensorflow/tensorflow/core/kernels/ |
| D | pack_op.cc | 40 class PackOp : public OpKernel { class
45 explicit PackOp(OpKernelConstruction* context) : OpKernel(context) { in PackOp() function in tensorflow::PackOp
126 PackOp<CPUDevice, type>)
144 PackOp<GPUDevice, type>)
160 PackOp<CPUDevice, int32>);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pack_op.cc | 39 class PackOp : public XlaOpKernel { class
41 explicit PackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in PackOp() function in tensorflow::__anone7c902130111::PackOp
89 REGISTER_XLA_OP(Name("Pack"), PackOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.h | 46 static TF::PackOp createMatMulOps(const std::vector<Value>& sliced_lhs,
|
| D | decompose_resource_ops.cc | 97 PackOp::getOperationName(), in DecomposeRngReadAndSkipOp()
190 Value new_state = rewriter.create<PackOp>(loc, res_type, pack_args); in matchAndRewrite()
|
| D | lower_tf.cc | 248 UnpackOp::getOperationName(), PackOp::getOperationName()}, in LowerDynamicStitchOp()
313 rewriter.replaceOpWithNewOp<PackOp>(op, op.getType(), values); in matchAndRewrite()
737 PackOp::getOperationName(), in LowerPackOp()
744 auto op = cast<PackOp>(src_op); in matchAndRewrite()
1286 PackOp::getOperationName(), in LowerResizeNearestNeighbor()
1393 auto out_shape = rewriter.create<PackOp>( in matchAndRewrite()
1430 rewriter.create<PackOp>(loc, in matchAndRewrite()
1466 rewriter.create<PackOp>(loc, in matchAndRewrite()
1490 rewriter.create<PackOp>(loc, in matchAndRewrite()
|
| D | unroll_batch_matmul.cc | 155 TF::PackOp ConvertTFBatchMatMulOp<BatchMatMulOpType>::createMatMulOps( in createMatMulOps()
185 return rewriter.create<TF::PackOp>(loc, packed_type, in createMatMulOps()
|
| D | shape_inference.cc | 346 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeInputsRequiredForOutput()
383 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeOutputComponent()
|
| D | legalize_hlo.cc | 337 Value start_indices_op = rewriter.create<PackOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 289 auto tf_pack_op = cast<TF::PackOp>(op); in matchAndRewrite()
297 rewriter.replaceOpWithNewOp<PackOp>(op, output_type, values, values_count, in matchAndRewrite()
|
| D | lower_static_tensor_list.cc | 239 rewriter.replaceOpWithNewOp<TF::PackOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1167 static LogicalResult Verify(PackOp op) { in Verify()
1493 : RewritePattern(PackOp::getOperationName(), 2, context) {} in RemoveRedundantUnpackPack()
1497 TFL::PackOp pack_op = cast<TFL::PackOp>(op); in matchAndRewrite()
1528 void PackOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 180 static LogicalResult Verify(PackOp op) { in Verify()
216 OpFoldResult PackOp::fold(ArrayRef<Attribute> operands) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1492 auto tf_pack_op = cast<TF::PackOp>(op); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2002 auto tfl_pack_op = cast<TFL::PackOp>(op); in matchAndRewrite()
|