Searched refs:kFirstPartitionInputIndex (Results 1 – 2 of 2) sorted by relevance
53 const int kFirstPartitionInputIndex = 3; variable80 return c->input(dimension + 1 + kFirstPartitionInputIndex) in GetRowPartitionTensor()83 return c->input(dimension + kFirstPartitionInputIndex).flat<INDEX_TYPE>(); in GetRowPartitionTensor()350 context->input(kFirstPartitionInputIndex); in GetFirstDimensionSize()375 context->input(kFirstPartitionInputIndex); in Compute()
39 constexpr int kFirstPartitionInputIndex = 3; variable147 return &context->tensors[node->inputs->data[kFirstPartitionInputIndex + 1 + in GetRowPartitionTensor()151 ->data[kFirstPartitionInputIndex + dimension]]; in GetRowPartitionTensor()622 context->tensors[node->inputs->data[kFirstPartitionInputIndex]]; in Eval()