Searched refs:input_backprop_flat (Results 1 – 2 of 2) sorted by relevance
343 auto input_backprop_flat = output->flat<T>(); in Compute() local346 int num_total_inputs = input_backprop_flat.size(); in Compute()355 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in Compute()
169 auto input_backprop_flat = input_backprop->flat<T>(); in SpatialMaxPoolWithArgMaxHelper() local177 EigenMatrixMap in_shard(input_backprop_flat.data() + in_start, 1, in SpatialMaxPoolWithArgMaxHelper()193 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in SpatialMaxPoolWithArgMaxHelper()