Lines Matching refs:positions
39 TfLiteTensor* positions = GetInput(context, node, kInputPositions);
41 // Only INT32 positions are supported.
42 TF_LITE_ENSURE_EQ(context, positions->type, kTfLiteInt32);
45 // TODO(mgubin): only 0D or 1D positions are currently supported.
46 TF_LITE_ENSURE(context, NumDimensions(positions) <= 1);
67 NumDimensions(input) + NumDimensions(positions) - 1;
74 for (int i = 0; i < positions->dims->size; ++i) {
75 output_shape->data[output_index++] = positions->dims->data[i];
85 TfLiteTensor* positions = GetInput(context, node, kInputPositions);
91 GetTensorData<index_type>(positions), GetTensorDims(positions), \
105 const int32* indexes = positions->data.i32;
107 for (int i = 0; i < positions->dims->data[0]; ++i) {
