Searched refs:isShapeTensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ | ||
| D | trt_shape_optimization_profiles.cc | 194 is_shape_tensor_[i] = input->isShapeTensor(); in SetShapeTensorMask() |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ | ||
| D | convert_nodes_test.cc | 387 bool isShapeTensor() const override { return false; } in isShapeTensor() function in tensorflow::tensorrt::convert::FakeITensor |
| D | convert_nodes.cc | 821 bool isShapeTensor() const override { return false; } in isShapeTensor() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor |