Searched refs:kDims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthtospace_op.cc | 76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
77 OP_REQUIRES(context, kDims == dims, in Compute()
78 errors::InvalidArgument("Input rank should be: ", kDims, in Compute()
112 auto Tinput = input.tensor<T, kDims>(); in Compute()
113 auto Toutput = outputs_tensor->tensor<T, kDims>(); in Compute()
|
| D | spacetodepth_op.cc | 76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
77 OP_REQUIRES(context, kDims == dims, in Compute()
78 errors::InvalidArgument("Input rank should be: ", kDims, in Compute()
114 auto Tinput = input.tensor<T, kDims>(); in Compute()
115 auto Toutput = outputs_tensor->tensor<T, kDims>(); in Compute()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 108 const int kDims = input_shape.dimensions_count(); in ExportFloatArray() local
110 kDims > 1 || (kDims == 1 && input_shape.dims(0) > 1)) { in ExportFloatArray()
111 for (int i = 0; i < kDims; ++i) { in ExportFloatArray()
2429 const int kDims = state_shape.dimensions_count(); in AddPlaceholderForRNNState() local
2430 for (int i = 0; i < kDims; ++i) { in AddPlaceholderForRNNState()
|