Home
last modified time | relevance | path

Searched defs:output_start_indices (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_cross_op.cc277 OutputUpdater(const std::vector<int64_t>& output_start_indices, in OutputUpdater()
712 std::vector<int64_t>* output_start_indices) { in CreateOutputTensors()
780 std::vector<int64_t> output_start_indices(batch_size); in Compute() local
851 std::vector<int64_t> output_start_indices(batch_size); in Compute() local
924 std::vector<int64_t> output_start_indices(batch_size); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc1334 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermute()
1356 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermuteStart()
H A Dhlo_instructions.cc1000 HloInstruction* output_start_indices, in HloCollectivePermuteInstruction()