Searched refs:MergeOp (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | control_flow_ops.cc | 201 MergeOp::MergeOp(OpKernelConstruction* context) : OpKernel(context) { in MergeOp() function in tensorflow::MergeOp
208 void MergeOp::Compute(OpKernelContext* context) { in Compute()
236 REGISTER_KERNEL_BUILDER(Name("Merge").Device(DEVICE_CPU), MergeOp);
238 Name("Merge").Device(DEVICE_DEFAULT).HostMemory("value_index"), MergeOp);
240 Name("Merge").Device(DEVICE_TPU_SYSTEM).HostMemory("value_index"), MergeOp);
241 REGISTER_KERNEL_BUILDER(Name("RefMerge").Device(DEVICE_CPU), MergeOp);
248 MergeOp);
255 MergeOp);
279 MergeOp); \
286 MergeOp)
|
| D | control_flow_ops.h | 67 class MergeOp : public OpKernel {
69 explicit MergeOp(OpKernelConstruction* context);
72 ~MergeOp() override {} in ~MergeOp()
74 TF_DISALLOW_COPY_AND_ASSIGN(MergeOp);
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/CodeGen/GlobalISel/ |
| D | RegBankSelect.cpp | 176 unsigned MergeOp; in repairReg() local
179 MergeOp = TargetOpcode::G_BUILD_VECTOR; in repairReg()
188 MergeOp = TargetOpcode::G_CONCAT_VECTORS; in repairReg()
191 MergeOp = TargetOpcode::G_MERGE_VALUES; in repairReg()
194 MIRBuilder.buildInstrNoInsert(MergeOp) in repairReg()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 570 LogicalResult MergeOp::verify() { in verify()
571 MergeOp merge = *this; in verify()
619 void MergeOp::print(OpAsmPrinter &p) { in print()
650 ParseResult MergeOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
|
| D | tf_executor_ops.td | 60 // and results. For example, MergeOp output type.
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| D | compatibility_analysis.cc | 77 mlir::tf_executor::MergeOp, mlir::tf_executor::EnterOp, in AnalyzeOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_device_propagation.cc | 81 tf_executor::IslandOp, tf_executor::MergeOp, in IsSupportedExecutorOp()
|
| D | tfg-to-tfe.cc | 346 rewriter.replaceOpWithNewOp<tf_executor::MergeOp>( in matchAndRewrite()
|
| D | shape_inference.cc | 212 tf_executor::LoopCondOp, tf_executor::MergeOp, in IsSupportedNonTFOp()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/CodeGen/GlobalISel/ |
| D | LegalizationArtifactCombiner.h | 232 static bool canFoldMergeOpcode(unsigned MergeOp, unsigned ConvertOp, in canFoldMergeOpcode() argument
235 switch (MergeOp) { in canFoldMergeOpcode()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 61 return mlir::isa<mlir::tf_executor::SwitchOp, mlir::tf_executor::MergeOp, in IsControlFlowV1Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1882 return builder_.create<mlir::tf_executor::MergeOp>(loc, types, operands, in CreateOperation()
|