Searched refs:CreateConcatOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/ |
| D | nnapi_hardware.cc | 78 std::unique_ptr<TargetHardwareOperation> CreateConcatOp() { in CreateConcatOp() function
88 TAC_REGISTER_NNAPI_OP(ConcatenationOp, CreateConcatOp);
89 TAC_REGISTER_NNAPI_OP(ReshapeOp, CreateConcatOp);
90 TAC_REGISTER_NNAPI_OP(PackOp, CreateConcatOp);
|
| D | cpu_hardware.cc | 156 std::unique_ptr<TargetHardwareOperation> CreateConcatOp() { in CreateConcatOp() function
165 TAC_REGISTER_CPU_OP(ConcatenationOp, CreateConcatOp);
166 TAC_REGISTER_CPU_OP(ReshapeOp, CreateConcatOp);
167 TAC_REGISTER_CPU_OP(PackOp, CreateConcatOp);
|
| D | gpu_hardware.cc | 137 std::unique_ptr<TargetHardwareOperation> CreateConcatOp() { in CreateConcatOp() function
198 TAC_REGISTER_GPU_OP(ConcatenationOp, CreateConcatOp);
199 TAC_REGISTER_GPU_OP(ReshapeOp, CreateConcatOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 98 mlir::TF::ConcatOp CreateConcatOp(const int concat_dimension, in CreateConcatOp() function
452 CreateConcatOp(concat_dimension, location, in HandleTileShardedOutputs()
|