Home
last modified time | relevance | path

Searched refs:PackParams (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpack.cc87 tflite::PackParams op_params; in PackImpl()
/external/tensorflow/tensorflow/stream_executor/
Dkernel.h526 void PackParams(KernelArgsArray<kNumberOfParameters> *args,
Dstream_executor_pimpl.h879 kernel.PackParams(&kernel_args, args...); in ThenLaunch()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h890 struct PackParams { struct
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h1057 tflite::PackParams op_params; in Pack()
1089 tflite::PackParams op_params; in Pack()
Dreference_ops.h1713 void Pack(const PackParams& params, const RuntimeShape* const* input_shapes, in Pack()
1774 void PackWithScaling(const PackParams& params, in PackWithScaling()