Searched refs:PaddedShapeFn (Results 1 – 3 of 3) sorted by relevance
55 typedef std::function<Status(const Tensor&, xla::Shape*)> PaddedShapeFn; typedef64 PaddedShapeFn padded_shape_fn, bool use_multiple_streams);75 const PaddedShapeFn& padded_shape_fn() const { return padded_shape_fn_; } in padded_shape_fn()84 PaddedShapeFn padded_shape_fn_;127 PaddedShapeFn padded_shape_fn;
143 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata()
67 tensorflow::XlaDevice::PaddedShapeFn shape_fn = in TFE_TensorHandleTensorDebugInfo()