Searched defs:sliced_input (Results 1 – 3 of 3) sorted by relevance
116 XlaOp sliced_input = in TopKWithPartitions() local152 XlaOp sliced_input = Slice(input, start_indices, limit_indices, strides); in TopKWithPartitions() local
32 XlaOp sliced_input = DynamicSlice(input, base_indices, window_sizes); in DynamicStridedSlice() local
109 ITensorProxyPtr sliced_input = slice_layer->getOutput(0); in NetworkWith2Inputs() local