Searched defs:output_slices (Results 1 – 4 of 4) sorted by relevance
63 const int64_t output_slices = channel_count; in Compute() local
249 output_slices, old_output_slices): argument
504 std::vector<Tensor*> output_slices(num_slices_); in ComputeInternal() local563 const std::vector<Tensor*>& output_slices) TF_ATTRIBUTE_NOINLINE { in SliceAndMaybePad()
417 std::vector<HloInstruction*> output_slices(concated_outputs.size() * in CreateFusedComputation() local