Searched defs:output_slice (Results 1 – 5 of 5) sorted by relevance
78 float* output_slice = in Compute() local
223 auto output_slice = output_shuffled.tensor<T, 5>().template chip<0>(i); in operator ()() local
132 float* output_slice = in Eval() local
541 void SetToConstant(Tensor* output_slice, in SetToConstant()549 Tensor* output_slice, const Device& device, const Tensor* input, in AssignFromInput()567 Tensor* output_slice = output_slices[i]; in SliceAndMaybePad() local
120 let output_slice = &mut output[output_index..(output_index + DECODED_BLOCK_LEN)]; localVariable