Searched refs:mlir_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.h | 40 llvm::Optional<tflite::BuiltinOperator> GetBuiltinOpCode(Operation *mlir_op);
46 Operation *mlir_op, uint32_t opcode_index,
|
| D | flatbuffer_import.cc | 1113 auto* mlir_op, in ConvertSubgraph()
1121 for (auto pair : llvm::enumerate(mlir_op->getResults())) { in ConvertSubgraph()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_cpu_op.h | 32 #define REGISTER_ALIASED_CPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
33 REGISTER_ALIASED_KERNEL(tf_op, mlir_op, CPU, input_type, output_type)
|
| D | base_gpu_op.h | 32 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
33 REGISTER_ALIASED_KERNEL(tf_op, mlir_op, GPU, input_type, output_type)
|
| D | build_defs.bzl | 142 inputs = [ctx.file.mlir_op, ctx.file._tfso],
148 "--input=%s" % ctx.file.mlir_op.path,
171 "mlir_op": attr.label(mandatory = True, allow_single_file = True),
245 mlir_op = "{op}_{platform}_{type}_{output_type}.mlir".format(
|
| D | base_op.h | 152 #define REGISTER_ALIASED_KERNEL(tf_op, mlir_op, platform, input_type, \ argument
158 MLIR_OP(mlir_op, platform, input_type, output_type));
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 258 #define MakeAndReturn(mlir_op) \ in ImportInstructionImpl() argument
261 func_builder->create<mlir::mhlo::mlir_op>(loc, result_type, operands, \ in ImportInstructionImpl()
737 #define NoAttributeCase(hlo_op_code, mlir_op) \ in ImportInstructionImpl() argument
739 MakeAndReturn(mlir_op); \ in ImportInstructionImpl()
|