HomeSort by relevance Sort by last modified time
    Searched refs:TRTEngineOpShapeInference (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/ops/
trt_engine_op.cc 27 extern Status TRTEngineOpShapeInference(InferenceContext* c);
38 .SetShapeFn(shape_inference::TRTEngineOpShapeInference);
  /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
trt_shfn.h 26 Status TRTEngineOpShapeInference(InferenceContext* c);
trt_shfn.cc 30 tensorflow::Status TRTEngineOpShapeInference(InferenceContext* context) {

Completed in 95 milliseconds