Lines Matching defs:feature_index
379 // feature_index operands be constants.
386 const HloInstruction* feature_index = custom_call->operand(6);
387 CHECK(feature_index->IsConstant());
388 int64 feature_index_value = feature_index->literal().Get<int64>({});
398 /*feature_index=*/feature_index_value,
410 const HloInstruction* feature_index = custom_call->operand(4);
411 CHECK(feature_index->IsConstant());
412 int64 feature_index_value = feature_index->literal().Get<int64>({});
426 /*feature_index=*/feature_index_value,
440 const HloInstruction* feature_index = custom_call->operand(6);
441 CHECK(feature_index->IsConstant());
442 int64 feature_index_value = feature_index->literal().Get<int64>({});
458 /*feature_index=*/feature_index_value,
