Searched refs:ConcatOp (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 100 class ConcatOp : public ConcatBaseOp { class
102 explicit ConcatOp(OpKernelConstruction* c) in ConcatOp() function in tensorflow::__anon8bb65f0c0111::ConcatOp
115 ConcatOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op.cc | 180 using ConcatOp = ConcatBaseOp<Device, T, NAME_IS_CONCAT_DIM>; typedef
189 ConcatOp<CPUDevice, type>) \
212 ConcatOp<GPUDevice, type>) \
233 ConcatOp<CPUDevice, int32>);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 154 auto start_position = rewriter->create<TF::ConcatOp>( in CreateSliceOpForTensorList()
163 auto slice_size = rewriter->create<TF::ConcatOp>( in CreateSliceOpForTensorList()
305 rewriter.replaceOpWithNewOp<TF::ConcatOp>( in matchAndRewrite()
466 auto list_shape = rewriter.create<TF::ConcatOp>( in matchAndRewrite()
542 rewriter.replaceOpWithNewOp<TF::ConcatOp>( in matchAndRewrite()
673 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 106 mlir::TF::ConcatOp* concat_op) { in CreateConcatOp()
136 *concat_op = builder->create<mlir::TF::ConcatOp>( in CreateConcatOp()
437 mlir::TF::ConcatOp concat_op; in HandleTileShardedOutputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 220 auto a1_concat_op = rewriter.create<tosa::ConcatOp>( in convertPackOp()
229 a1_concat_op = rewriter.create<tosa::ConcatOp>(op->getLoc(), concat_type, in convertPackOp()
687 auto concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op()
722 concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 944 OpT, ConcatOp, ConcatV2Op>::value>::type * = nullptr>
949 int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1; in Verify()
961 void ConcatOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/IR/ |
| D | Shape.cpp | 343 OpFoldResult ConcatOp::fold(ArrayRef<Attribute> operands) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2804 rewriter->create<shape::ConcatOp>(loc, result_batch_shape, tail_shape); in BroadcastBatchMatMulV2Operands()
|
| /external/llvm-project/llvm/lib/CodeGen/SelectionDAG/ |
| D | DAGCombiner.cpp | 18476 SDValue ConcatOp = VecOp.getOperand(Elt / ConcatNumElts); in visitEXTRACT_VECTOR_ELT() local
18479 ConcatOp, NewIdx); in visitEXTRACT_VECTOR_ELT()
|