Searched refs:setBroadcastAcrossBatch (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ | ||
| D | convert_nodes_test.cc | 231 void setBroadcastAcrossBatch(bool broadcastAcrossBatch) override {} in setBroadcastAcrossBatch() function in tensorflow::tensorrt::convert::FakeITensor |
| D | convert_nodes.cc | 668 void setBroadcastAcrossBatch(bool broadcastAcrossBatch) override {} in setBroadcastAcrossBatch() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor |