Home
last modified time | relevance | path

Searched refs:addInputs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_tf_framework.cc41 conversion.addInputs(OpKernelContextType::get(rewriter.getContext())); in matchAndRewrite()
43 conversion.addInputs(arg_type.index(), arg_type.value()); in matchAndRewrite()
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DEagerOperationBuilderTest.java55 public void addInputs() { in addInputs() method in EagerOperationBuilderTest
/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DDecomposeCallGraphTypes.cpp74 conversion.addInputs(argType.index(), decomposedTypes); in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Transforms/
DDialectConversion.h67 void addInputs(unsigned origInputNo, ArrayRef<Type> types);
71 void addInputs(ArrayRef<Type> types);
/external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/
DConvertGPUToSPIRV.cpp189 signatureConverter.addInputs(argType.index(), convertedType); in lowerAsEntryFunction()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc440 sig_conversion.addInputs(arg.getArgNumber(), new_type); in matchAndRewrite()
444 sig_conversion.addInputs({MemRefType::get(result_type.getShape(), in matchAndRewrite()
Dlegalize_to_linalg.cc988 signature_converter.addInputs( in matchAndRewrite()
1371 signatureConverter.addInputs(0, src_type.getElementType()); in matchAndRewrite()
1372 signatureConverter.addInputs(1, src_type.getElementType()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Transforms/Utils/
DDialectConversion.cpp2342 void TypeConverter::SignatureConversion::addInputs(unsigned origInputNo, in addInputs() function in TypeConverter::SignatureConversion
2346 addInputs(types); in addInputs()
2351 void TypeConverter::SignatureConversion::addInputs(ArrayRef<Type> types) { in addInputs() function in TypeConverter::SignatureConversion
2469 result.addInputs(inputNo, convertedTypes); in convertSignatureArg()
/external/llvm-project/mlir/docs/
DDialectConversion.md385 void addInputs(unsigned origInputNo, ArrayRef<Type> types);
389 void addInputs(ArrayRef<Type> types);
/external/llvm-project/mlir/lib/Dialect/SPIRV/
DSPIRVLowering.cpp486 signatureConverter.addInputs(argType.index(), convertedType); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc955 signature_conversion.addInputs( in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp239 result.addInputs(en.index(), converted); in convertFunctionSignature()