Home
last modified time | relevance | path

Searched refs:CustomCallArgEncodingSet (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Dpasses.h45 class CustomCallArgEncodingSet; variable
71 std::function<void(CustomCallArgEncodingSet&)> populate_arg_encodings;
H A Drt_to_llvm.cc220 CustomCallOp op, CustomCallArgEncodingSet &encodings, Globals &g, in EncodeArguments()
322 CustomCallArgEncodingSet &arg_encoding, in CustomCallOpLowering()
370 CustomCallArgEncodingSet &arg_encoding_;
458 CustomCallArgEncodingSet args = DefaultArgEncodings(); in runOnOperation()
H A Dcustom_call_encoding.h91 class CustomCallArgEncodingSet {
102 CustomCallArgEncodingSet &Add() { in Add()
496 CustomCallArgEncodingSet DefaultArgEncodings();
H A Dcompilation_pipeline.h44 std::function<void(CustomCallArgEncodingSet&)> populate_arg_encodings;
H A Dcustom_call_encoding.cc54 using EncodedArg = CustomCallArgEncodingSet::Encoded;
56 FailureOr<EncodedArg> CustomCallArgEncodingSet::Encode(Globals &g, in Encode()
937 CustomCallArgEncodingSet DefaultArgEncodings() { in DefaultArgEncodings()
938 CustomCallArgEncodingSet encodings; in DefaultArgEncodings()