Home
last modified time | relevance | path

Searched refs:padded_shape_fn (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h64 PaddedShapeFn padded_shape_fn, bool use_multiple_streams);
75 const PaddedShapeFn& padded_shape_fn() const { return padded_shape_fn_; } in padded_shape_fn() function
127 PaddedShapeFn padded_shape_fn; member
Dxla_device.cc143 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata() argument
148 padded_shape_fn_(std::move(padded_shape_fn)), in Metadata()
198 options.padded_shape_fn ? options.padded_shape_fn in XlaDevice()
/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc68 xla_device->metadata().padded_shape_fn(); in TFE_TensorHandleTensorDebugInfo()