Searched refs:int32_output (Results 1 – 2 of 2) sorted by relevance
81 Tensor int32_output(DT_INT32, params.forward_output_shape()); in Compute() local85 SpatialAvgPool<Device, int32>(context, &int32_output, int32_input, params, in Compute()89 output->flat<T>() = int32_output.flat<int32>() in Compute()
328 DataType int32_output =340 int32_output));