Lines Matching refs:feature_index
323 // feature_index operands be constants.
330 const HloInstruction* feature_index = custom_call->operand(6);
331 CHECK(feature_index->IsConstant());
332 int64 feature_index_value = feature_index->literal().Get<int64>({});
342 /*feature_index=*/feature_index_value,
354 const HloInstruction* feature_index = custom_call->operand(4);
355 CHECK(feature_index->IsConstant());
356 int64 feature_index_value = feature_index->literal().Get<int64>({});
370 /*feature_index=*/feature_index_value,
384 const HloInstruction* feature_index = custom_call->operand(6);
385 CHECK(feature_index->IsConstant());
386 int64 feature_index_value = feature_index->literal().Get<int64>({});
402 /*feature_index=*/feature_index_value,
