Searched refs:shape_fn (Results 1 – 2 of 2) sorted by relevance
24 srcs = ["shape_fn/trt_shfn.cc"],25 hdrs = ["shape_fn/trt_shfn.h"],
67 tensorflow::XlaDevice::PaddedShapeFn shape_fn = in TFE_TensorHandleTensorDebugInfo() local70 status->status = shape_fn(*tensor, &padded_shape); in TFE_TensorHandleTensorDebugInfo()