Home
last modified time | relevance | path

Searched defs:ConvertEinsum (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Deinsum.cc665 class ConvertEinsum : public OpConverterBase<ConvertEinsum> { class
667 explicit ConvertEinsum(OpConverterParams* params) in ConvertEinsum() function in tensorflow::tensorrt::convert::__anonc98924470111::ConvertEinsum
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc3380 TEST_P(OpConverter_FP32_Test, ConvertEinsum) { in TEST_P() argument