Home
last modified time | relevance | path

Searched defs:kDims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dspacetodepth_op.cc76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
Ddepthtospace_op.cc76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc108 const int kDims = input_shape.dimensions_count(); in ExportFloatArray() local
2429 const int kDims = state_shape.dimensions_count(); in AddPlaceholderForRNNState() local