Searched refs:ConcatV2Op (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 108 class ConcatV2Op : public ConcatBaseOp { class
110 explicit ConcatV2Op(OpKernelConstruction* c) in ConcatV2Op() function in tensorflow::__anon8bb65f0c0111::ConcatV2Op
119 ConcatV2Op);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op.cc | 182 using ConcatV2Op = ConcatBaseOp<Device, T, NAME_IS_AXIS>; typedef
194 ConcatV2Op<CPUDevice, type>)
217 ConcatV2Op<GPUDevice, type>)
240 ConcatV2Op<CPUDevice, int32>);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 944 OpT, ConcatOp, ConcatV2Op>::value>::type * = nullptr>
980 class HoistCwiseUnaryOutOfConcat : public OpRewritePattern<TF::ConcatV2Op> {
983 : OpRewritePattern<TF::ConcatV2Op>(context) {} in HoistCwiseUnaryOutOfConcat()
984 LogicalResult matchAndRewrite(TF::ConcatV2Op op,
989 TF::ConcatV2Op op, PatternRewriter &rewriter) const { in matchAndRewrite()
1014 rewriter.create<ConcatV2Op>(loc, op.getType(), unary_ops_args, op.axis()); in matchAndRewrite()
1053 class HoistCwiseBinaryOutOfConcat : public OpRewritePattern<TF::ConcatV2Op> {
1056 : OpRewritePattern<TF::ConcatV2Op>(context) {} in HoistCwiseBinaryOutOfConcat()
1057 LogicalResult matchAndRewrite(TF::ConcatV2Op op,
1077 TF::ConcatV2Op op, int64_t axis,
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 137 return rewriter.create<ConcatV2Op>(loc, type, ValueRange(vals), axis); in ValuesToRank1()
738 {ConstOp::getOperationName(), ConcatV2Op::getOperationName(), in LowerPackOp()
769 rewriter.replaceOpWithNewOp<ConcatV2Op>(op, op.getType(), expanded_inputs, in matchAndRewrite()
807 ConcatV2Op::getOperationName(), in LowerSpaceToBatchNDOp()
868 auto full_paddings = rewriter.create<ConcatV2Op>( in matchAndRewrite()
|
| D | collection_ops_util.cc | 68 return builder.create<TF::ConcatV2Op>( in GetIndicesForElement()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1412 TF::ConcatV2Op expanded_fft_len = rewriter.create<TF::ConcatV2Op>( in matchAndRewrite()
|
| D | legalize_tf.cc | 222 auto tf_concat_op = cast<TF::ConcatV2Op>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 635 auto tf_concatv2_op = cast<TF::ConcatV2Op>(op); in matchAndRewrite()
|