Home
last modified time | relevance | path

Searched refs:setBroadcastAcrossBatch (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc353 void setBroadcastAcrossBatch(bool broadcastAcrossBatch) override {} in setBroadcastAcrossBatch() function in tensorflow::tensorrt::convert::FakeITensor
Dconvert_nodes.cc789 void setBroadcastAcrossBatch(bool broadcastAcrossBatch) override {} in setBroadcastAcrossBatch() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor